Skip to content

Commit 94ef17c

Browse files
authored
Update model_checkpoint.py (#7625)
1 parent 92cf396 commit 94ef17c

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

pytorch_lightning/callbacks/model_checkpoint.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -420,14 +420,16 @@ def __init_triggers(
420420
# Default to running once after each validation epoch if neither
421421
# every_n_train_steps nor every_n_val_epochs is set
422422
if every_n_train_steps is None and every_n_val_epochs is None and train_time_interval is None:
423-
self._every_n_val_epochs = 1
424-
self._every_n_train_steps = 0
423+
every_n_val_epochs = 1
424+
every_n_train_steps = 0
425425
log.debug("Both every_n_train_steps and every_n_val_epochs are not set. Setting every_n_val_epochs=1")
426426
else:
427-
self._every_n_val_epochs = every_n_val_epochs or 0
428-
self._every_n_train_steps = every_n_train_steps or 0
427+
every_n_val_epochs = every_n_val_epochs or 0
428+
every_n_train_steps = every_n_train_steps or 0
429429

430430
self._train_time_interval: Optional[timedelta] = train_time_interval
431+
self._every_n_val_epochs: int = every_n_val_epochs
432+
self._every_n_train_steps: int = every_n_train_steps
431433

432434
# period takes precedence over every_n_val_epochs for backwards compatibility
433435
if period is not None:

0 commit comments

Comments
 (0)