yolov5 : Can i add more epochs on the model already done custom training?

Question

I'm trying to make image detection model with 30,000 images. I just tried 5 epochs which took about 5 hours of training.

The question is, if I want to simply add more epochs on the same model, how can i do that? Is there a way that I can load custom trained model?

Since it takes so long, I wish I can continue training from 5 epochs already done training. I'm using colab, and I experienced a few times loosing connection during the training so I'm afraid of giving more epochs from the start of training.

Answer 1

Yes, you can use the standard yolo training command and provide the path to your previously training model in the -- weights paramater.

!python train.py --img 960 --batch 20 --epochs 10 --data dataset.yaml --weights your_model.pt --cache --exist-ok