From 0697dd306d578f0cd3c4d23b768da0106e54a095 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 2 Jul 2020 07:18:58 -0400 Subject: [PATCH] Fixes #2455 (#2463) --- pytorch_lightning/trainer/training_loop.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pytorch_lightning/trainer/training_loop.py b/pytorch_lightning/trainer/training_loop.py index 3ee4973053..aa41087d29 100644 --- a/pytorch_lightning/trainer/training_loop.py +++ b/pytorch_lightning/trainer/training_loop.py @@ -527,6 +527,7 @@ class TrainerTrainLoopMixin(ABC): def run_training_epoch_end(self, epoch_output): model = self.get_model() if self.is_overridden('training_epoch_end', model=model): + self.global_step += 1 epoch_output = model.training_epoch_end(epoch_output) _processed_outputs = self.process_output(epoch_output) log_epoch_metrics = _processed_outputs[2]