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

Save best model weights instead of the last #13

Closed

Erlemar opened this issue Oct 4, 2020 · 0 comments · Fixed by #37

Labels

enhancement good first issue

Projects

Pipeline development

Owner

Erlemar commented Oct 4, 2020

Currently in train.py the model saved has weights from the last epoch. It would be better to load the best checkpoint and save model from it instead.

https://pytorch-lightning.readthedocs.io/en/latest/weights_loading.html

Erlemar added enhancement good first issue labels

Erlemar linked a pull request

that will close this issue

Merged

Erlemar closed this as completed in #37

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