Revert "Correct online_lr overriding"

This reverts commit d27bf33.
SJTU-DMTai · Apr 13, 2024 · 344a604 · 344a604
1 parent d27bf33
commit 344a604
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/qlib/contrib/meta/incremental/model.py b/qlib/contrib/meta/incremental/model.py
@@ -235,9 +235,9 @@ def _init_meta_optimizer(self, lr_da=0.01, lr_x=None, lr_y=None, **kwargs):
     def override_online_lr_(self):
         if self.online_lr is not None:
             if 'lr_model' in self.online_lr:
-                self.framework.opt.param_groups[0]['lr'] = self.online_lr['lr_model']
+                self.lr_model = self.online_lr['lr_model']
             if 'lr_ma' in self.online_lr:
-                self.lr_model = self.online_lr['lr_ma']
+                self.framework.opt.param_groups[0]['lr'] = self.online_lr['lr_ma']
             if 'lr_da' in self.online_lr:
                 self.opt.param_groups[0]['lr'] = self.online_lr['lr_da']
             else: