update

Lightning-AI · Sep 16, 2021 · e9a46d0 · e9a46d0
1 parent 3668fed
commit e9a46d0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pytorch_lightning/loops/dataloader/dataloader_loop.py b/pytorch_lightning/loops/dataloader/dataloader_loop.py
@@ -57,6 +57,8 @@ def reset(self) -> None:
         """Resets the internal state."""
         if not self.restarting:
             self.dataloader_progress.current.reset()
+        else:
+            self.dataloader_progress.current.reset_on_restart()
 
     def on_advance_start(self, *args: Any, **kwargs: Any) -> None:
         self.dataloader_progress.increment_ready()