From e9a46d0ef7c8445b044f5a71a1a710689321fef3 Mon Sep 17 00:00:00 2001 From: tchaton Date: Thu, 16 Sep 2021 14:38:07 +0100 Subject: [PATCH] update --- pytorch_lightning/loops/dataloader/dataloader_loop.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/pytorch_lightning/loops/dataloader/dataloader_loop.py b/pytorch_lightning/loops/dataloader/dataloader_loop.py index 6b5fecd07e807..00a5ee32b933b 100644 --- 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()