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

moved epoch state increment to before save #13

Merged
merged 1 commit into from
May 8, 2022
Merged

moved epoch state increment to before save #13

merged 1 commit into from
May 8, 2022

Conversation

B-lanc
Copy link

@B-lanc B-lanc commented May 8, 2022

saving should be done after incrementing the state["epoch"]
beforehand, if you were to run train for 1 epoch at a time, and then save and reload, it will always be on epoch 0. There is no real issue since the epoch state is actually not being used at all, but were it be used for something, it will create problems.

@f90
Copy link
Owner

f90 commented May 8, 2022

Good point, thanks! Merging...

@f90 f90 merged commit 86c113d into f90:master May 8, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants