update

Author: deependujha

.github/workflows/ci-tests-pytorch.yml

@@ -75,7 +75,7 @@ jobs:
           - { os: "macOS-14", pkg-name: "pytorch", python-version: "3.12", pytorch-version: "2.7" }
           - { os: "ubuntu-22.04", pkg-name: "pytorch", python-version: "3.12", pytorch-version: "2.7" }
           - { os: "windows-2022", pkg-name: "pytorch", python-version: "3.12", pytorch-version: "2.7" }
-    timeout-minutes: 50
+    timeout-minutes: 60
     env:
       PACKAGE_NAME: ${{ matrix.pkg-name }}
       TORCH_URL: "https://download.pytorch.org/whl/cpu/"

src/lightning/pytorch/loops/loop.py

@@ -41,7 +41,7 @@ def restarting(self, restarting: bool) -> None:
 
     @property
     def is_resuming(self) -> bool:
-        """Whether we're resuming training from a checkpoint."""
+        """Indicates whether training is being resumed from a checkpoint."""
         return self._resuming_from_checkpoint
 
     def reset_restart_stage(self) -> None:

src/lightning/pytorch/loops/training_epoch_loop.py

@@ -237,10 +237,10 @@ def reset(self) -> None:
 
     def on_run_start(self, data_fetcher: _DataFetcher) -> None:
         # `iter()` was called once in `FitLoop.setup_data()` already
-        # Only call `iter()` if all following cases:
-        # 1. Not restarting
-        # 2. Not resuming from checkpoint (not is_resuming)
-        # 3. Past first epoch (current_epoch > 0)
+        # Call `iter()` again only when:
+        #       1. Not restarting
+        #       2. Not resuming from checkpoint (not is_resuming)
+        #       3. Past first epoch (current_epoch > 0)
         if (self.trainer.current_epoch > 0 and not self.trainer.fit_loop.is_resuming) and not self.restarting:
             iter(data_fetcher)  # creates the iterator inside the fetcher