ModelCheckpoint save nothing #10327
Answered
by
Ryuk17
Ryuk17
asked this question in
Lightning Trainer API: Trainer, LightningModule, LightningDataModule
-
I want to use ModelCheckpoint to save mode while training, however, nothing has been saved. The following is my code. I don't know what leads to this problem, any suggestions? |
Beta Was this translation helpful? Give feedback.
Answered by
Ryuk17
Nov 3, 2021
Replies: 1 comment
-
I solve the problem by add |
Beta Was this translation helpful? Give feedback.
0 replies
Answer selected by
Ryuk17
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
I solve the problem by add
self.log('val_loss', val_loss)
in functionvalidation_epoch_end
.