user shceduler itself to update model parameter

marrlab · Nov 2, 2023 · 1af128b · 1af128b
1 parent d605670
commit 1af128b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/domainlab/algos/trainers/train_fbopt_b.py b/domainlab/algos/trainers/train_fbopt_b.py
@@ -116,7 +116,8 @@ def set_model_with_mu(self):
         """
         set model multipliers
         """
-        self.model.hyper_update(epoch=None, fun_scheduler=HyperSetter(self.hyper_scheduler.mmu))
+        # self.model.hyper_update(epoch=None, fun_scheduler=HyperSetter(self.hyper_scheduler.mmu))
+        self.model.hyper_update(epoch=None, fun_scheduler=self.hyper_scheduler)
 
     def tr_epoch(self, epoch, flag_info=False):
         """