Hello,
Can i train ssd-mobilnet on google colab and use it on jetson nano? Can i use jetson-inference on google colab for training?
How can i do it?
Thank you.
The different training scripts from jetson-inference are Git submodules, so you could try running just those submodules on Google Colab. I haven’t confirmed on Colab, but I have done training on my Linux PC using the same code:
- GitHub - dusty-nv/pytorch-ssd: MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in PyTorch. Out-of-box support for retraining on Open Images dataset. ONNX and Caffe2 support. Experiment Ideas like CoordConv.
- GitHub - dusty-nv/pytorch-classification: Training of image classification models with PyTorch
You will need to make sure the prerequisites are installed in the cloud environment.
1 Like
What are the prerequisites required?
PyTorch along with the packages in requirements.txt