Skip to content

Commit 213921e

Browse files
committed
Fixes #2455
1 parent 1a40963 commit 213921e

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

pytorch_lightning/trainer/training_loop.py

+1
Original file line numberDiff line numberDiff line change
@@ -527,6 +527,7 @@ def run_on_epoch_end_hook(self, model):
527527
def run_training_epoch_end(self, epoch_output):
528528
model = self.get_model()
529529
if self.is_overridden('training_epoch_end', model=model):
530+
self.global_step += 1
530531
epoch_output = model.training_epoch_end(epoch_output)
531532
_processed_outputs = self.process_output(epoch_output)
532533
log_epoch_metrics = _processed_outputs[2]

0 commit comments

Comments
 (0)