How to convert a Trained Model in COLAB to be used in Jetson NANO?

Hello everyone! I’m a beginner in Jetson NANO. Currently I’m using COLAB to train some Detection models, but the trained model in COLAB is in Pytorch (.pth extension). I’d like to convert this model to ONNX in order to use it on the Jetson Nano.
Thanks.

Hi,

You can use the PyTorch API to convert a .pth file into .onnx directly.
Below is a tutorial for your reference:

https://pytorch.org/docs/stable/onnx.html

Thanks.

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.