From 1af128b34dfbb6427b15f2f3c8c39f8864cc5f37 Mon Sep 17 00:00:00 2001 From: smilesun Date: Thu, 2 Nov 2023 17:24:00 +0100 Subject: [PATCH] user shceduler itself to update model parameter --- domainlab/algos/trainers/train_fbopt_b.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/domainlab/algos/trainers/train_fbopt_b.py b/domainlab/algos/trainers/train_fbopt_b.py index a82201382..e40c09b54 100644 --- 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): """