pacs diva

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