Training updating model in many cycles of training #2879
Unanswered
johanenforcer
asked this question in
Q&A
Replies: 1 comment 3 replies
-
@johanenforcer you can start training from previously trained models with no code changes, i.e.: $ python train.py --epochs 300 --weights yolov5s.pt
$ python train.py --epochs 300 --weights runs/train/exp/weights/best.pt
$ python train.py --epochs 300 --weights runs/train/exp2/weights/best.pt
$ python train.py --epochs 300 --weights runs/train/exp3/weights/best.pt
$ python train.py --epochs 300 --weights runs/train/exp4/weights/best.pt
$ etc... Each training run will start from the previous training runs best checkpoint. |
Beta Was this translation helpful? Give feedback.
3 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
Right now, I have made the training procedure in many cycles of training, meaning that I am rerunning the training process but with the previously trained model as the next model to train in the many cycles of training. My problem is that I am not quite sure exactly what I need to change in order to make this fully work. What I have done currently is that I load the model variable as the current model like so:
The most important thing is that the model variable gets updated but am I missing updating other important variables also, like ckpt maybe or is doing what I have been doing fine?
Thanks in advance!
Beta Was this translation helpful? Give feedback.
All reactions