fix issue #507

marrlab · Nov 7, 2023 · fbefd73 · fbefd73
1 parent 1af128b
commit fbefd73
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/domainlab/algos/trainers/fbopt_mu_controller.py b/domainlab/algos/trainers/fbopt_mu_controller.py
@@ -252,3 +252,6 @@ def update_setpoint(self, epo_reg_loss, epo_task_loss):
         update setpoint
         """
         return self.set_point_controller.observe(epo_reg_loss, epo_task_loss)
+
+    def __call__(self, epoch):
+        return self.mmu