Fixes #2455 (#2463)

2020-07-02 07:18:58 -04:00 · 2020-07-02 07:18:58 -04:00 · 0697dd306d
parent afdfba1dc6
commit 0697dd306d
1 changed files with 1 additions and 0 deletions
--- 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]