Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fixes #2455 (#2463)

This commit is contained in:
William Falcon 2020-07-02 07:18:58 -04:00 committed by GitHub
parent afdfba1dc6
commit 0697dd306d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pytorch_lightning/trainer/training_loop.py
View File

@ -527,6 +527,7 @@ class TrainerTrainLoopMixin(ABC):
def run_training_epoch_end(self, epoch_output): def run_training_epoch_end(self, epoch_output):
model = self.get_model() model = self.get_model()
if self.is_overridden('training_epoch_end', model=model): if self.is_overridden('training_epoch_end', model=model):
self.global_step += 1
epoch_output = model.training_epoch_end(epoch_output) epoch_output = model.training_epoch_end(epoch_output)
_processed_outputs = self.process_output(epoch_output) _processed_outputs = self.process_output(epoch_output)
log_epoch_metrics = _processed_outputs[2] log_epoch_metrics = _processed_outputs[2]