Is it possible to save best model only ONE time after training process? #13122
Unanswered
GillianGrayson
asked this question in
Lightning Trainer API: Trainer, LightningModule, LightningDataModule
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
Hi!
Is it possible to save the best model only ONE time after training process (
EarlyStopping
callback)?If I understand correctly,
ModelCheckpoint
allows you to save the best model only in iterative mode.For example, if
save_top_k=1
, then when a new best model is found, the checkpoint will overwrite on the hard drive.But what if I want to reduce the load on the hard drive and save the final best model only ONE time at the very end of training?
The parameter
save_last=True
will not work, because the last state of the model will not be optimal (I am usingEarlyStopping
withpatience=100
).Beta Was this translation helpful? Give feedback.
All reactions