Skip to content

Commit 701888b

Browse files
developer0hyeyhkwon-DT01
authored andcommitted
fix typo
This code is not working now. I could run with this code. Signed-off-by: yhkwon-DT01 <[email protected]>
1 parent 6442fce commit 701888b

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)