Update model_checkpoint.py

Lightning-AI · Mar 5, 2021 · d91636d · d91636d
1 parent 725719b
commit d91636d
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
@@ -210,8 +210,7 @@ def on_train_batch_end(self, trainer, pl_module, outputs, batch, batch_idx, data
         if self._should_skip_saving_checkpoint(trainer):
             return
         step = trainer.global_step
-        skip_batch = self.every_n_batches < 1 or ((step + 1) % self.every_n_batches != 0)
-        log.warning(f"in on_train_batch_end at step {step}, every_n_batches={self.every_n_batches}, going to skip batch? {skip_batch}")
+        skip_batch = self.every_n_steps < 1 or ((step + 1) % self.every_n_steps != 0)
         if skip_batch:
             return
         self.save_checkpoint(trainer, pl_module)