Skip to content

Commit 25211b1

Browse files
authored
Merge pull request #336 from developer0hye/patch-2
fix typo in example code
2 parents d279f05 + 701888b commit 25211b1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

py/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ data = torch.randn((1, 3, 224, 224)).to("cuda")
2121
traced_model = torch.jit.trace(model, [data])
2222

2323
# Compile module
24-
compiled_trt_model = trtorch.compile(model, {
24+
compiled_trt_model = trtorch.compile(traced_model, {
2525
"input_shapes": [data.shape],
2626
"op_precision": torch.half, # Run in FP16
2727
})

0 commit comments

Comments
 (0)