How can I save the best model with a minimun epoch? #6499
Unanswered
xianyuanliu
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
-
Hi all,
In my case, I hope model training for 10 epochs before adding an extra loss, and I need to get the best model for the rest epochs. Can I use
ModelCheckpointto do it? Many thanks!Beta Was this translation helpful? Give feedback.
All reactions