Skip to content

Commit 725719b

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

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

pytorch_lightning/callbacks/model_checkpoint.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -211,6 +211,7 @@ def on_train_batch_end(self, trainer, pl_module, outputs, batch, batch_idx, data
211211
return
212212
step = trainer.global_step
213213
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}")
214215
if skip_batch:
215216
return
216217
self.save_checkpoint(trainer, pl_module)

0 commit comments

Comments
 (0)