Update args_fbopt.py

marrlab · Oct 10, 2023 · 75fae54 · 75fae54
1 parent ddde147
commit 75fae54
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/domainlab/algos/trainers/args_fbopt.py b/domainlab/algos/trainers/args_fbopt.py
@@ -25,6 +25,9 @@ def add_args2parser_fbopt(parser):
 
     parser.add_argument('--coeff_ma_output_state', type=float, default=0.5,
                         help='setpoint output as state exponential moving average')
+
+    parser.add_argument('--coeff_ma_setpoint', type=float, default=0.5,
+                        help='setpoint average')
 
 
     parser.add_argument('--exp_shoulder_clip', type=float, default=10,