Merge pull request #1273 from davinnovation/patch-3

Fix typo in README
pytorch · Aug 16, 2022 · e8c971a · e8c971a
2 parents 9bb0087 + 99b655c
commit e8c971a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -80,7 +80,8 @@ trt_ts_module = torch_tensorrt.compile(torch_script_module,
             # For static size shape=[1, 3, 224, 224]
             dtype=torch.half) # Datatype of input tensor. Allowed options torch.(float|half|int8|int32|bool)
     ],
-    enabled_precisions = {torch.half}, # Run with FP16)
+    enabled_precisions = {torch.half}, # Run with FP16
+)
 
 result = trt_ts_module(input_data) # run inference
 torch.jit.save(trt_ts_module, "trt_torchscript_module.ts") # save the TRT embedded Torchscript