Skip to content

Commit f7f2ce4

Browse files
committed
Update model_checkpoint.py
1 parent 094eddb commit f7f2ce4

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
@@ -210,6 +210,7 @@ def on_train_batch_end(self, trainer, pl_module, outputs, batch, batch_idx, data
210210
return
211211
step = trainer.global_step
212212
skip_batch = self.every_n_batches < 1 or ((step + 1) % self.every_n_batches != 0)
213+
log.warning(f"in on_train_batch_end at step {step}, every_n_batches={self.every_n_batches}, going to skip batch? {skip_batch}")
213214
if skip_batch:
214215
return
215216
self.save_checkpoint(trainer, pl_module)

0 commit comments

Comments
 (0)