-
Notifications
You must be signed in to change notification settings - Fork 501
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
How to re-traning when stop #142
Comments
You can modify the training code by adding
before
This will restore the weights from the checkpoint file, but do be aware of the learning rate as the original learning rate decays in the later epochs, so by restarting the training from epoch 0 with the already trained weights, you might have to make some modifications. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
I have run code "train.py". But I stop it.
I wonder if I can how to re-train with the latest checkpoint?
Please help me.
The text was updated successfully, but these errors were encountered: