Skip to content

Commit 5984247

Browse files
ananthsubawaelchli
andauthored
Apply suggestions from code review
Co-authored-by: Adrian Wälchli <[email protected]>
1 parent 9888e03 commit 5984247

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

pytorch_lightning/callbacks/base.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -104,6 +104,7 @@ def on_train_epoch_end(
104104
"""Called when the train epoch ends.
105105
106106
To access all batch outputs at the end of the epoch, either:
107+
107108
1. Implement `training_epoch_end` in the `LightningModule` and access outputs via the module OR
108109
2. Cache data across train batch hooks inside the callback implementation to post-process in this hook.
109110
"""

pytorch_lightning/core/hooks.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -240,6 +240,7 @@ def on_train_epoch_end(self, unused: Optional = None) -> None:
240240
Called in the training loop at the very end of the epoch.
241241
242242
To access all batch outputs at the end of the epoch, either:
243+
243244
1. Implement `training_epoch_end` in the LightningModule OR
244245
2. Cache data across steps on the attribute(s) of the `LightningModule` and access them in this hook
245246
"""

0 commit comments

Comments
 (0)