if you are using Jetson Nano, you can directly run the model trained in PyTorch on the Nano.
If you want to speed the model up on Nano, you can use something like https://github.com/NVIDIA-AI-IOT/torch2trt 58 to offload part of the model to TensorRT