From c81165f854f1c8568816f6a845415fb24e2a2141 Mon Sep 17 00:00:00 2001
From: smilesun <smilesun.east@gmail.com>
Date: Fri, 6 Oct 2023 17:05:29 +0200
Subject: [PATCH] pacs diva

---
 .../benchmark/benchmark_fbopt_pacs_diva.yaml  | 64 +++++++++++++++++++
 1 file changed, 64 insertions(+)
 create mode 100644 examples/benchmark/benchmark_fbopt_pacs_diva.yaml

diff --git a/examples/benchmark/benchmark_fbopt_pacs_diva.yaml b/examples/benchmark/benchmark_fbopt_pacs_diva.yaml
new file mode 100644
index 000000000..23b5d425a
--- /dev/null
+++ b/examples/benchmark/benchmark_fbopt_pacs_diva.yaml
@@ -0,0 +1,64 @@
+mode: grid
+
+output_dir: zoutput/benchmarks/benchmark_fbopt_pacs_full
+
+sampling_seed: 0
+
+startseed: 0
+endseed: 2
+
+test_domains:
+  - sketch
+
+domainlab_args:
+  tpath: examples/tasks/task_pacs_path_list.py
+  dmem: False
+  lr: 5e-5
+  epos: 200
+  es: 10
+  bs: 64
+  san_check: True
+  npath: examples/nets/resnet50domainbed.py
+  npath_dom: examples/nets/resnet50domainbed.py
+  npath_topic_distrib_img2topic: examples/nets/resnet50domainbed.py
+  npath_encoder_sandwich_layer_img2h4zd: examples/nets/resnet50domainbed.py
+  exp_shoulder_clip: 10
+  mu_clip: 10_000
+  coeff_ma: 0.5
+  zx_dim: 0
+  zy_dim: 64
+  zd_dim: 64
+
+
+
+
+Shared params:
+  ini_setpoint_ratio:
+    min: 0.9
+    max: 0.99
+    num: 3
+    distribution: uniform
+
+  k_i_gain:
+    min: 0.0001
+    max: 0.1
+    num: 3
+    distribution: uniform
+
+  mu_init:
+    min: 0.0001
+    max: 1.0
+    num: 3
+    distribution: loguniform
+
+
+
+# Test fbopt with different hyperparameter configurations
+
+diva_fbopt:
+  aname: diva
+  trainer: fbopt
+  shared:
+    - ini_setpoint_ratio
+    - k_i_gain
+    - mu_init