How to Resume Yolov3 training?

514 Views Asked by At

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.

1

There are 1 best solutions below

1
On BEST ANSWER

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