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.

Sign up for GitHub

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

Jump to bottom

Fix epoch number for saving a training checkpoint #281

Merged

tanaysoni merged 1 commit into master from fix-epoch-for-checkpoint

Mar 17, 2020

Contributor

tanaysoni commented Mar 17, 2020

Trainer uses self.from_epoch to get the current epoch number when saving a checkpoint.

Currently, from_epoch gets updated at the end of the train loop, thus it lacks behind by 1.

This PR makes a change to update the from_epoch at the beginning of the train loop.


          Fix epoch number for saving checkpoint

a22897f

tanaysoni requested a review from tholor

March 17, 2020 09:11

tholor approved these changes

View reviewed changes

tanaysoni merged commit 58d93bf into master

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