docs: fix order of on_fit_start() hook (#16180)

Fixes https://github.com/Lightning-AI/lightning/issues/16170
2022-12-23 07:06:57 +11:00 · 2022-12-23 07:06:57 +11:00 · 94e6d52b7e
parent f08bc67e4d
commit 94e6d52b7e
1 changed files with 1 additions and 1 deletions
--- 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