.

marrlab · Oct 9, 2023 · 9253f8c · 9253f8c
1 parent 72074dc
commit 9253f8c
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 3 deletions.
diff --git a/examples/benchmark/benchmark_fbopt_mnist_jigen.yaml b/examples/benchmark/benchmark_fbopt_mnist_jigen.yaml
@@ -53,6 +53,14 @@ Shared params:
     num: 3
     distribution: uniform
 
+  gamma_reg:
+    min: 0.01
+    max: 1000
+    num: 10
+    distribution: loguniform
+
+
+
 # Test fbopt with different hyperparameter configurations
 
 jigen_feedback:
@@ -67,9 +75,15 @@ jigen_feedback:
 
 jigen_feedforward:
   aname: jigen
-  trainer: fbopt
+  trainer: hyperscheduler
   shared:
-    - pperm
-
+    - gamma_reg
+
+jigen_fixed_penalty:
+  aname: jigen
+  trainer: basic
+  shared:
+    - gamma_reg
+
 erm:
   aname: deepall
diff --git a/examples/benchmark/benchmark_fbopt_pacs_jigen.yaml b/examples/benchmark/benchmark_fbopt_pacs_jigen.yaml
@@ -77,3 +77,6 @@ jigen_fixed_penalty:
   trainer: basic
   shared:
     - gamma_reg
+
+erm:
+  aname: deepall