Skip to content

Commit 99b655c

Browse files
Fix typo in README
1 parent 9bb0087 commit 99b655c

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

README.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,8 @@ trt_ts_module = torch_tensorrt.compile(torch_script_module,
8080
# For static size shape=[1, 3, 224, 224]
8181
dtype=torch.half) # Datatype of input tensor. Allowed options torch.(float|half|int8|int32|bool)
8282
],
83-
enabled_precisions = {torch.half}, # Run with FP16)
83+
enabled_precisions = {torch.half}, # Run with FP16
84+
)
8485
8586
result = trt_ts_module(input_data) # run inference
8687
torch.jit.save(trt_ts_module, "trt_torchscript_module.ts") # save the TRT embedded Torchscript

0 commit comments

Comments
 (0)