Skip to content

Commit d91636d

Browse files
committed
Update model_checkpoint.py
1 parent 725719b commit d91636d

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

pytorch_lightning/callbacks/model_checkpoint.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -210,8 +210,7 @@ def on_train_batch_end(self, trainer, pl_module, outputs, batch, batch_idx, data
210210
if self._should_skip_saving_checkpoint(trainer):
211211
return
212212
step = trainer.global_step
213-
skip_batch = self.every_n_batches < 1 or ((step + 1) % self.every_n_batches != 0)
214-
log.warning(f"in on_train_batch_end at step {step}, every_n_batches={self.every_n_batches}, going to skip batch? {skip_batch}")
213+
skip_batch = self.every_n_steps < 1 or ((step + 1) % self.every_n_steps != 0)
215214
if skip_batch:
216215
return
217216
self.save_checkpoint(trainer, pl_module)

0 commit comments

Comments
 (0)