Update trainer.py

Lightning-AI · May 4, 2021 · d3ef271 · d3ef271
1 parent 8928822
commit d3ef271
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py
@@ -1214,9 +1214,9 @@ def _cache_logged_metrics(self):
  def call_hook(self, hook_name: str, *args, **kwargs) -> Any:
  # Note this implementation is copy/pasted into the TrainLoop class
  # in TrainLoop._on_train_epoch_end_hook
- # This was done to manage the deprecation of an argument to
- # on_train_epoch_end
- # If making chnages to this function, ensure that those changes are also made to
+ # This was done to manage the deprecation of the `outputs` argument to
+ # `on_train_epoch_end`.
+ # If making changes to this function, ensure that those changes are also made to
  # TrainLoop._on_train_epoch_end_hook
 
  # set hook_name to model + reset Result obj