Retrain ssd mobilnet

Hello iam using jetson inference to train object detection model. I did the training for 60 epochs and its complete.
Now i want to add more pictures to the dataset and retrain the model.
Is there a way to retrain the model rather than starting training from scratch?
Thank you


Yes. Please use the --pretrained-ssd configuration to input a pre-trained model.

$ python3 --pretrained-ssd=models/mobilenet-v1-ssd-mp-0_675.pth ...

For more configure support, you can check this file directly.


1 Like