Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Cannot resume train.py from checkpoint #1

Open
jackhftang opened this issue Sep 2, 2017 · 1 comment
Open

Cannot resume train.py from checkpoint #1

jackhftang opened this issue Sep 2, 2017 · 1 comment
Assignees

Comments

@jackhftang
Copy link
Collaborator

No description provided.

@flyyufelix
Copy link

First, you have to save/export the weights. You can either use model.save_weights("model.h5") or use checkpoint (see https://machinelearningmastery.com/check-point-deep-learning-models-keras/).

Once you have the exported weight you can resume training by calling the Keras fit() function with initial_epoch parameter set to the number of epoch which the training left off. For example, if you paused the training at epoch 10 and want to resume training afterwards, you can set initial_epoch at 10 (see https://keras.io/models/model/#fit).

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

3 participants