From 99b655cdae066c6b376b7d3e9333ec51d7eb4d01 Mon Sep 17 00:00:00 2001 From: davinnovation Date: Tue, 16 Aug 2022 23:29:13 +0900 Subject: [PATCH] Fix typo in README --- README.md | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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