How do I save model/continue training?

Question

How do I save model/continue training?

Closed this issue 6 years ago · 2 comments

I don't know if this is normal, but my model building takes more than an hour and takes around 3 hours to get the model.h5. Is it possible to save the training progress and continue it later on?

Answer 1 · 2019-07-08T07:59:07.000Z

Yes, in train.py there should be a flag CONTINUE_TRAIN near the top that you can set to True.

Answer 2 · 2019-07-08T19:23:12.000Z

Ahh thank you. Should've checked the code before opening an issue. It worked great btw. Thanks for sharing your code. Keep up your good work! Thanks