From 3d1e1a73d5208fbbf28ec1f1204f0e60b84e0886 Mon Sep 17 00:00:00 2001 From: Mickus Timothee Date: Fri, 31 May 2024 13:25:31 +0300 Subject: [PATCH] optimizer global step removal --- mammoth/utils/optimizers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mammoth/utils/optimizers.py b/mammoth/utils/optimizers.py index d3f29664..1c4b08f3 100644 --- a/mammoth/utils/optimizers.py +++ b/mammoth/utils/optimizers.py @@ -354,7 +354,7 @@ def state_dict(self): """Returns the state dictionary""" return { 'optimizers': {k: v.state_dict() for k, v in self.suboptimizers.items()}, - 'steps': self._steps, + # 'steps': self._steps, } def load_state_dict(self, state_dict): @@ -372,7 +372,7 @@ def load_state_dict(self, state_dict): else: logger.info("Some components do not match. Do not load optimizer from checkpoint.") - self._steps = state_dict["steps"] + # self._steps = state_dict["steps"] @property def training_step(self):