Update model_checkpoint.py

Lightning-AI · Mar 2, 2021 · 094eddb · 094eddb
1 parent 0ec9949
commit 094eddb
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/pytorch_lightning/callbacks/model_checkpoint.py b/pytorch_lightning/callbacks/model_checkpoint.py
@@ -218,10 +218,9 @@ def on_validation_end(self, trainer, pl_module):
         """
         checkpoints can be saved at the end of the val loop
         """
-        epoch = trainer.current_epoch
         skip = (
             self._should_skip_saving_checkpoint(trainer) or self.every_n_epochs < 1
-            or (epoch + 1) % self.every_n_epochs != 0
+            or (trainer.current_epoch + 1) % self.every_n_epochs != 0
         )
         if skip:
             return