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
Add SaveCheckpoint
callback for checkpointing
#2317
Conversation
BaseModelClass.save
BaseModelClass.save
ModelCheckpoint
for compatibility with BaseModelClass.save
Could be useful to add a post-training cleanup method in |
Codecov ReportAttention:
Additional details and impacted files@@ Coverage Diff @@
## main #2317 +/- ##
==========================================
+ Coverage 89.26% 89.29% +0.03%
==========================================
Files 145 145
Lines 11827 11872 +45
==========================================
+ Hits 10557 10601 +44
- Misses 1270 1271 +1 ☔ View full report in Codecov by Sentry. |
ModelCheckpoint
for compatibility with BaseModelClass.save
SaveCheckpoint
callback for checkpointing
docs/release_notes/index.md
file if fixing a bug or adding a new feature.on-merge: backport to x.x.x
label.To do