We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9bb0087 commit 99b655cCopy full SHA for 99b655c
README.md
@@ -80,7 +80,8 @@ trt_ts_module = torch_tensorrt.compile(torch_script_module,
80
# For static size shape=[1, 3, 224, 224]
81
dtype=torch.half) # Datatype of input tensor. Allowed options torch.(float|half|int8|int32|bool)
82
],
83
- enabled_precisions = {torch.half}, # Run with FP16)
+ enabled_precisions = {torch.half}, # Run with FP16
84
+)
85
86
result = trt_ts_module(input_data) # run inference
87
torch.jit.save(trt_ts_module, "trt_torchscript_module.ts") # save the TRT embedded Torchscript
0 commit comments