Skip to content

Commit 630db82

Browse files
docs: move on_fit_end hook to teardown stage. (#21143)
* docs: move `on_fit_end` hook to `teardown` stage. * Apply suggestion from @bhimrazy --------- Co-authored-by: Bhimraj Yadav <[email protected]>
1 parent 06bed20 commit 630db82

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

docs/source-pytorch/common/hooks.rst

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -213,13 +213,14 @@ with the source of each hook indicated:
213213
│ ├── [LightningModule]
214214
│ └── [Strategy]
215215
216-
├── on_fit_end()
217-
│ ├── [Callbacks]
218-
│ ├── [LightningModule]
219-
│ └── [Strategy]
220-
221216
└── teardown(stage="fit")
222-
└── [Callbacks only]
217+
├── [Strategy]
218+
├── on_fit_end()
219+
│ ├── [Callbacks]
220+
│ └── [LightningModule]
221+
├── [LightningDataModule]
222+
├── [Callbacks]
223+
└── [LightningModule]
223224
224225
***********************
225226
Testing Loop Hook Order

0 commit comments

Comments
 (0)