diff --git a/README.md b/README.md index 78789a9451..9cd32c9533 100644 --- 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