Skip to content

Commit

Permalink
Merge branch 'release/1.2.2' of https://github.com/kaushikb11/pytorch…
Browse files Browse the repository at this point in the history
…-lightning into release/1.2.2
  • Loading branch information
tchaton committed Mar 3, 2021
2 parents 1f3cba0 + 692f096 commit ed420fe
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pytorch_lightning/trainer/training_loop.py
Original file line number Diff line number Diff line change
Expand Up @@ -296,7 +296,7 @@ def training_step(self, split_batch, batch_idx, opt_idx, hiddens):
self.trainer.logger_connector.cache_logged_metrics()

self._check_training_step_output(training_step_output)

training_step_output = self.trainer.call_hook("training_step_end", training_step_output)

training_step_output_for_epoch_end, training_step_output = self._process_training_step_output(
Expand Down

0 comments on commit ed420fe

Please sign in to comment.