From 9253f8c373e6c2c2aa14e324e6a94a610be6e0d4 Mon Sep 17 00:00:00 2001
From: smilesun <smilesun.east@gmail.com>
Date: Mon, 9 Oct 2023 10:32:57 +0200
Subject: [PATCH] .

---
 .../benchmark_fbopt_mnist_jigen.yaml          | 20 ++++++++++++++++---
 .../benchmark/benchmark_fbopt_pacs_jigen.yaml |  3 +++
 2 files changed, 20 insertions(+), 3 deletions(-)

diff --git a/examples/benchmark/benchmark_fbopt_mnist_jigen.yaml b/examples/benchmark/benchmark_fbopt_mnist_jigen.yaml
index c0f3e25e5..d371a8129 100644
--- 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
index c96d1704b..953d47427 100644
--- 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