Skip to content

Commit 81ea4d5

Browse files
carmoccaBorda
authored andcommitted
Clarify default epoch and step values in ModelCheckpoint (#16651)
(cherry picked from commit 679f0db)
1 parent 4554e76 commit 81ea4d5

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/pytorch_lightning/callbacks/model_checkpoint.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,8 @@ class ModelCheckpoint(Checkpoint):
8080
... filename='{epoch}-{val_loss:.2f}-{other_metric:.2f}'
8181
... )
8282
83-
By default, filename is ``None`` and will be set to ``'{epoch}-{step}'``.
83+
By default, filename is ``None`` and will be set to ``'{epoch}-{step}'``, where "epoch" and "step" match
84+
the number of finished epoch and optimizer steps respectively.
8485
monitor: quantity to monitor. By default it is ``None`` which saves a checkpoint only for the last epoch.
8586
verbose: verbosity mode. Default: ``False``.
8687
save_last: When ``True``, saves an exact copy of the checkpoint to a file `last.ckpt` whenever a checkpoint

0 commit comments

Comments
 (0)