Cosine schedule fix typos

bghira · Oct 31, 2023 · 558f6b7 · 558f6b7
1 parent 2a45762
commit 558f6b7
Showing 1 changed file with 2 additions and 10 deletions.
diff --git a/helpers/training/custom_schedule.py b/helpers/training/custom_schedule.py
@@ -210,14 +210,6 @@ def step(self, epoch=None):
                 self.T_cur = epoch
         self.last_epoch = epoch
 
-        with _enable_get_lr_call(self):
-            for i, data in enumerate(zip(self.optimizer.param_groups, self.get_lr())):
-                param_group, lr = data
-                param_group['lr'] = lr
-                self.print_lr(self.verbose, i, lr, epoch)
-
-        self._last_lr = [group['lr'] for group in self.optimizer.param_groups]
-
         class _enable_get_lr_call:
             def __init__(self, o):
                 self.o = o
@@ -233,7 +225,7 @@ def __exit__(self, type, value, traceback):
         with _enable_get_lr_call(self):
             for i, data in enumerate(zip(self.optimizer.param_groups, self.get_lr())):
                 param_group, lr = data
-                param_group["lr"] = lr
+                param_group['lr'] = lr
                 self.print_lr(self.verbose, i, lr, epoch)
 
-        self._last_lr = [group["lr"] for group in self.optimizer.param_groups]
+        self._last_lr = [group['lr'] for group in self.optimizer.param_groups]