I am trying to make an app that trains a YOLO model, and after every epoch saves the statistics somewhere (to be sent to another part of the app), and then continue training the model in the next epoch.
I don’t really understand how to make that. For example, I do:
model.train(data='...', epochs=1)
That will train it for 1 epoch, but how do I train the same for 1 more epoch? Does the argument epochs have to be equal to 2? Do I do resume=true
? Do I load the model trained on the previous epoch from ‘weightsbest.pt’ or ‘weighslast.pt’?
vct12345 is a new contributor to this site. Take care in asking for clarification, commenting, and answering.
Check out our Code of Conduct.