We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6442fce commit 701888bCopy full SHA for 701888b
py/README.md
@@ -21,7 +21,7 @@ data = torch.randn((1, 3, 224, 224)).to("cuda")
21
traced_model = torch.jit.trace(model, [data])
22
23
# Compile module
24
-compiled_trt_model = trtorch.compile(model, {
+compiled_trt_model = trtorch.compile(traced_model, {
25
"input_shapes": [data.shape],
26
"op_precision": torch.half, # Run in FP16
27
})
0 commit comments