Update fbopt_setpoint_ada.py

marrlab · Oct 10, 2023 · bff815b · bff815b
1 parent 75fae54
commit bff815b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/domainlab/algos/trainers/fbopt_setpoint_ada.py b/domainlab/algos/trainers/fbopt_setpoint_ada.py
@@ -41,7 +41,7 @@ def __init__(self, state=None, args=None):
                 state = DominateAnyComponent()
         self.transition_to(state)
         self.ma_epo_reg_loss = None
-        self.coeff_ma = 0.5  # FIXME
+        self.coeff_ma = args.coeff_ma_setpoint 
         self.state_task_loss = 0.0
         self.state_epo_reg_loss = [0.0 for _ in range(10)] # FIXME
         self.coeff_ma_output = args.coeff_ma_output_state