diff --git a/docs/source-pytorch/common/hooks.rst b/docs/source-pytorch/common/hooks.rst
index 4c20ad1b857f0..89c1c15d0413f 100644
--- a/docs/source-pytorch/common/hooks.rst
+++ b/docs/source-pytorch/common/hooks.rst
@@ -83,13 +83,30 @@ with the source of each hook indicated:
     trainer.fit()
     │
     ├── setup(stage="fit")
-    │   └── [Callbacks only]
-    │
-    ├── on_fit_start()
+    │   ├── [LightningDataModule]
     │   ├── [Callbacks]
     │   ├── [LightningModule]
+    │   ├── [LightningModule.configure_shared_model()]
+    │   ├── [LightningModule.configure_model()]
+    │   ├── Strategy.restore_checkpoint_before_setup
+    │   │   ├── [LightningModule.on_load_checkpoint()]
+    │   │   ├── [LightningModule.load_state_dict()]
+    │   │   ├── [LightningDataModule.load_state_dict()]
+    │   │   ├── [Callbacks.on_load_checkpoint()]
+    │   │   └── [Callbacks.load_state_dict()]
     │   └── [Strategy]
     │
+    ├── on_fit_start()
+    │   ├── [Callbacks]
+    │   └── [LightningModule]
+    │
+    ├── Strategy.restore_checkpoint_after_setup
+    │   ├── [LightningModule.on_load_checkpoint()]
+    │   ├── [LightningModule.load_state_dict()]
+    │   ├── [LightningDataModule.load_state_dict()]
+    │   ├── [Callbacks.on_load_checkpoint()]
+    │   └── [Callbacks.load_state_dict()]
+    │
     ├── on_sanity_check_start()
     │   ├── [Callbacks]
     │   ├── [LightningModule]