From 94e6d52b7e2f2a9ffc21f7e11e087808666fe710 Mon Sep 17 00:00:00 2001
From: David Gilbertson <gilbertson.david@gmail.com>
Date: Fri, 23 Dec 2022 07:06:57 +1100
Subject: [PATCH] docs: fix order of on_fit_start() hook (#16180)

Fixes https://github.com/Lightning-AI/lightning/issues/16170
---
 docs/source-pytorch/common/lightning_module.rst | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docs/source-pytorch/common/lightning_module.rst b/docs/source-pytorch/common/lightning_module.rst
index cd64307ad9..690b0209c4 100644
--- a/docs/source-pytorch/common/lightning_module.rst
+++ b/docs/source-pytorch/common/lightning_module.rst
@@ -1295,9 +1295,9 @@ for more information.
 
     def train_on_device(model):
         # called PER DEVICE
-        on_fit_start()
         setup("fit")
         configure_optimizers()
+        on_fit_start()
 
         # the sanity check runs here