Skip to content

Commit

Permalink
Correct a typo in <pytorch-cat-dog.md> file
Browse files Browse the repository at this point in the history
  • Loading branch information
marco-cheung committed Jun 7, 2023
1 parent 2d53107 commit 34b896a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/pytorch-cat-dog.md
Original file line number Diff line number Diff line change
Expand Up @@ -107,7 +107,7 @@ Note that the models are saved under `jetson-inference/python/training/classific

## Converting the Model to ONNX

To run our re-trained ResNet-18 model with TensorRT for testing and realtime inference, first we need to convert the PyTorch model into <a href="https://onnx.ai/">ONNX format</a> format so that TensorRT can load it. ONNX is an open model format that supports many of the popular ML frameworks, including PyTorch, TensorFlow, TensorRT, and others, so it simplifies transferring models between tools.
To run our re-trained ResNet-18 model with TensorRT for testing and realtime inference, first we need to convert the PyTorch model into <a href="https://onnx.ai/">ONNX format</a> so that TensorRT can load it. ONNX is an open model format that supports many of the popular ML frameworks, including PyTorch, TensorFlow, TensorRT, and others, so it simplifies transferring models between tools.

PyTorch comes with built-in support for exporting PyTorch models to ONNX, so run the following command to convert our Cat/Dog model with the provided `onnx_export.py` script:

Expand Down

0 comments on commit 34b896a

Please sign in to comment.