From d724465ae6dff77fc1f89ed4ebb12069ce86e501 Mon Sep 17 00:00:00 2001 From: Danielle Pintz <38207072+daniellepintz@users.noreply.github.com> Date: Mon, 21 Feb 2022 14:50:10 -0500 Subject: [PATCH] Fix docstrings of on_fit_start/end #12016 --- pytorch_lightning/trainer/callback_hook.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pytorch_lightning/trainer/callback_hook.py b/pytorch_lightning/trainer/callback_hook.py index 28c799e0f0..1e7f6f8c9a 100644 --- a/pytorch_lightning/trainer/callback_hook.py +++ b/pytorch_lightning/trainer/callback_hook.py @@ -117,7 +117,7 @@ class TrainerCallbackHookMixin(ABC): .. deprecated:: v1.6 `TrainerCallbackHookMixin.on_fit_start` was deprecated in v1.6 and will be removed in v1.8. - Called when the trainer initialization begins, model has not yet been set. + Called when fit begins. """ rank_zero_deprecation( "`TrainerCallbackHookMixin.on_fit_start` was deprecated in v1.6 and will be removed in v1.8." @@ -130,7 +130,7 @@ class TrainerCallbackHookMixin(ABC): .. deprecated:: v1.6 `TrainerCallbackHookMixin.on_fit_end` was deprecated in v1.6 and will be removed in v1.8. - Called when the trainer initialization begins, model has not yet been set. + Called when fit ends. """ rank_zero_deprecation( "`TrainerCallbackHookMixin.on_fit_end` was deprecated in v1.6 and will be removed in v1.8."