Skip to content

Commit 856cd66

Browse files
committed
fix teardown bug
1 parent fc2d612 commit 856cd66

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytorch_lightning/loops/epoch/training_epoch_loop.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -227,7 +227,7 @@ def on_run_end(self) -> List[List[STEP_OUTPUT]]:
227227

228228
def teardown(self) -> None:
229229
"""Frees memory of tracked epoch outputs."""
230-
self.epoch_output = None
230+
self._epoch_output = None
231231

232232
def _on_train_epoch_end_hook(self, processed_epoch_output: List[List[STEP_OUTPUT]]) -> None:
233233
"""Runs ``on_train_epoch_end hook``."""

0 commit comments

Comments
 (0)