How to save a checkpoint every n steps and overwrite the previous saved checkpoints? #6329
-
|
How do I save checkpoints every, as well as deleting and/or overwriting the previously saved checkpoints? |
Beta Was this translation helpful? Give feedback.
Replies: 1 comment 2 replies
-
|
Just to be sure, do you mean every n step or every n epoch. checkpoint = pl.callbacks.ModelCheckpoint(period=n) |
Beta Was this translation helpful? Give feedback.
-
|
Update, when this PR #6146 gets merged it will be possible to save checkpoint every n training step |
Beta Was this translation helpful? Give feedback.
-
|
Thank you for the response, I'll be eagerly awaiting the next version |
Beta Was this translation helpful? Give feedback.
Just to be sure, do you mean every n step or every n epoch.
For doing it every n epoch, you can initialize model checkpoint with
periodparameter