Skip to content

Commit 4142d3f

Browse files
authored
typo fix in doc on saving models (#2818)
1 parent 6cad83d commit 4142d3f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

docsrc/user_guide/saving_models.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ Here's an example usage
3535
inputs = [torch.randn((1, 3, 224, 224)).cuda()]
3636
# trt_ep is a torch.fx.GraphModule object
3737
trt_gm = torch_tensorrt.compile(model, ir="dynamo", inputs)
38-
torchtrt.save(trt_gm, "trt.ep", inputs=inputs)
38+
torch_tensorrt.save(trt_gm, "trt.ep", inputs=inputs)
3939
4040
# Later, you can load it and run inference
4141
model = torch.export.load("trt.ep").module()
@@ -97,4 +97,4 @@ Here's an example usage
9797
# file_path can be trt.ep or trt.ts file obtained via saving the model (refer to the above section)
9898
inputs = [torch.randn((1, 3, 224, 224)).cuda()]
9999
model = torch_tensorrt.load(<file_path>).module()
100-
model(*inputs)
100+
model(*inputs)

0 commit comments

Comments
 (0)