yoloresuming-training

How to Resume Yolov3 training?


I am new to deep learning, I have a yolov3 model that I have been training on my custom data. Every time I train, the training seems to start from scratch. How do I make the model continue its training from where it stopped last time? The setup I have is the same as this repo.


Solution

  • You can use model.load_weights(path_to_checkpoint) just after the model is defined at line 41 in train.py and continue training where you left off