Skip to content

Commit 0a2405f

Browse files
committed
chore: Doc updates
Signed-off-by: Dheeraj Peri <[email protected]>
1 parent e7bbc2d commit 0a2405f

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

docsrc/tutorials/creating_torchscript_module_in_python.rst

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
Creating a TorchScript Module
44
------------------------------
55

6-
Once you have a trained model you want to compile with TRTorch, you need to start by converting that model from Python code to TorchScript code.
6+
Once you have a trained model you want to compile with Torch-TensorRT, you need to start by converting that model from Python code to TorchScript code.
77
PyTorch has detailed documentation on how to do this https://pytorch.org/tutorials/beginner/Intro_to_TorchScript_tutorial.html but briefly here is the
88
here is key background information and the process:
99

@@ -82,12 +82,12 @@ include these components. We can run the script compiler on our LeNet module by
8282
model = LeNet()
8383
script_model = torch.jit.script(model)
8484
85-
There are reasons to use one path or another, the PyTorch documentation has information on how to choose. From a TRTorch prespective, there is
85+
There are reasons to use one path or another, the PyTorch documentation has information on how to choose. From a Torch-TensorRT prespective, there is
8686
better support (i.e your module is more likely to compile) for traced modules because it doesn't include all the complexities of a complete
8787
programming language, though both paths supported.
8888

8989
After scripting or tracing your module, you are given back a TorchScript Module. This contains the code and parameters used to run the module stored
90-
in a intermediate representation that TRTorch can consume.
90+
in a intermediate representation that Torch-TensorRT can consume.
9191

9292
Here is what the LeNet traced module IR looks like:
9393

0 commit comments

Comments
 (0)