From fbefd73591040b84bb92c2cb16c44bdd77f772ed Mon Sep 17 00:00:00 2001
From: smilesun <smilesun.east@gmail.com>
Date: Tue, 7 Nov 2023 15:42:34 +0100
Subject: [PATCH] fix issue #507

---
 domainlab/algos/trainers/fbopt_mu_controller.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/domainlab/algos/trainers/fbopt_mu_controller.py b/domainlab/algos/trainers/fbopt_mu_controller.py
index 3193ae225..2429a3bd6 100644
--- 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