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 25075e2 commit bf5fa71Copy full SHA for bf5fa71
docsrc/user_guide/saving_models.rst
@@ -34,7 +34,7 @@ Here's an example usage
34
model = MyModel().eval().cuda()
35
inputs = [torch.randn((1, 3, 224, 224)).cuda()]
36
# trt_ep is a torch.fx.GraphModule object
37
- trt_gm = torch_tensorrt.compile(model, ir="dynamo", inputs)
+ trt_gm = torch_tensorrt.compile(model, ir="dynamo", inputs=inputs)
38
torch_tensorrt.save(trt_gm, "trt.ep", inputs=inputs)
39
40
# Later, you can load it and run inference
0 commit comments