I trained a model on my Jetson Nano using the tutorial from Jetson-inference (Hello AI World). However, now I want to train a model on Google Colab instead of my Jetson Nano. Is there a tutorial available to guide me through this process using Google Colab?
Hi @jardel.vieira, I don’t have a tutorial specifically on using Google Colab, but if you are able to run PyTorch in it and install the other dependencies of train_ssd.py (like pandas), then you should be able to. I do regularly train the models on PC with the same code that runs on the Jetson. You can just clone the pytorch-ssd submodule instead of the whole jetson-inference repo:
Then when you’ve trained it, you can export the ONNX, and copy the ONNX (and labels.txt) to your Jetson and proceed normally.
This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.