Apply suggestions from code review

Author: Borda

pytorch_lightning/callbacks/base.py

@@ -104,7 +104,7 @@ def on_train_epoch_end(
         """Called when the train epoch ends.
 
         To access all batch outputs at the end of the epoch, either:
-        1. Implement `training_epoch_end` in the LightningModule and access outputs via the module OR
+        1. Implement `training_epoch_end` in the `LightningModule` and access outputs via the module OR
         2. Cache data across train batch hooks inside the callback implementation to post-process in this hook.
         """
         pass

pytorch_lightning/core/hooks.py

@@ -241,8 +241,7 @@ def on_train_epoch_end(self, unused: Optional = None) -> None:
 
         To access all batch outputs at the end of the epoch, either:
         1. Implement `training_epoch_end` in the LightningModule OR
-        2. Cache data across steps on attribute(s) of the LightningModule
-           and access them in this hook
+        2. Cache data across steps on the attribute(s) of the `LightningModule` and access them in this hook
         """
 
     def on_validation_epoch_start(self) -> None: