From 530caef0a730a76360527b72566a7e78c944fff2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Adrian=20W=C3=A4lchli?= Date: Thu, 2 Sep 2021 18:37:15 +0200 Subject: [PATCH] Remove redundant `optimizer_idx` reset on epoch (#9285) --- pytorch_lightning/trainer/progress.py | 1 - 1 file changed, 1 deletion(-) diff --git a/pytorch_lightning/trainer/progress.py b/pytorch_lightning/trainer/progress.py index bcd2426da3e6a..8085bb7d2f75a 100644 --- a/pytorch_lightning/trainer/progress.py +++ b/pytorch_lightning/trainer/progress.py @@ -193,7 +193,6 @@ def optimizer_steps(self) -> int: def reset_on_epoch(self) -> None: self.optimizer.reset_on_epoch() - self.optimizer_idx = 0 def load_state_dict(self, state_dict: dict) -> None: self.optimizer.load_state_dict(state_dict["optimizer"])