Add independent flag to track checkpoint resumption.

Signed-off-by: sudipto baral <sudiptobaral.me@gmail.com>

Author: sudiptob2

src/lightning/pytorch/loops/fit_loop.py

@@ -207,7 +207,7 @@ def skip(self) -> bool:
     @property
     def _is_resuming(self) -> bool:
         """Whether we're resuming training from a checkpoint."""
-        return self._loaded_from_state_dict
+        return self._resuming_from_checkpoint
 
     def run(self) -> None:
         self.setup_data()

src/lightning/pytorch/loops/loop.py

@@ -23,6 +23,7 @@ class _Loop:
     def __init__(self, trainer: "pl.Trainer") -> None:
         self._restarting = False
         self._loaded_from_state_dict = False
+        self._resuming_from_checkpoint = False
         self.trainer = trainer
 
     @property
@@ -87,6 +88,7 @@ def load_state_dict(
                 v.load_state_dict(state_dict.copy(), prefix + k + ".")
         self.restarting = True
         self._loaded_from_state_dict = True
+        self._resuming_from_checkpoint = True
 
     def _load_from_state_dict(self, state_dict: dict, prefix: str) -> None:
         for k, v in self.__dict__.items():
@@ -102,4 +104,5 @@ def _load_from_state_dict(self, state_dict: dict, prefix: str) -> None:
     def on_iteration_done(self) -> None:
         self._restarting = False
         self._loaded_from_state_dict = False
+        self._resuming_from_checkpoint = False
         self.reset_restart_stage()