File tree Expand file tree Collapse file tree 2 files changed +2
-0
lines changed Expand file tree Collapse file tree 2 files changed +2
-0
lines changed Original file line number Diff line number Diff 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 """
Original file line number Diff line number Diff 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 """
You can’t perform that action at this time.
0 commit comments