Pytorch-lightning: Managing Checkpoints

Created on 12 Feb 2020  Â·  3Comments  Â·  Source: PyTorchLightning/pytorch-lightning

🚀 Feature

During training there is support for checkpointing multiple variants of the model. However to utilize these checkpoints you need to parse the file names to get out information. It would be nice if lightning either wrote out statistics to make it easy to find the right model, or even better allowed you to load based on validation statistics.

load_from_checkpoint(ckpt, metric=‘f-score’, mode=‘max’)
enhancement good first issue help wanted won't fix

Most helpful comment

Well, I would maybe rather add a method that would return a path to such checkpoint and then you can load it on your own...

find_checkpoint(folder, metric=‘f-score’, mode=‘max’)

cc: @PyTorchLightning/core-contributors thoughts?

All 3 comments

sounds interesting, @srush would you be interested in submitting such PR?

Well, I would maybe rather add a method that would return a path to such checkpoint and then you can load it on your own...

find_checkpoint(folder, metric=‘f-score’, mode=‘max’)

cc: @PyTorchLightning/core-contributors thoughts?

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.

Was this page helpful?
0 / 5 - 0 ratings