Merge branch 'fbopt' into fbopt_output_ma

marrlab · Oct 10, 2023 · ef48ee0 · ef48ee0
2 parents 03a6720 + 0669d54
commit ef48ee0
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 10 deletions.
diff --git a/README.md b/README.md
@@ -37,6 +37,8 @@ then
 
 #### Guide for Helmholtz GPU cluster
 ```
+conda create --name domainlab_py39 python=3.9
+conda activate domainlab_py39
 conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.6 -c pytorch -c conda-forge
 conda install torchmetric==0.10.3
 pip install -r requirements_notorch.txt 

diff --git a/examples/benchmark/benchmark_fbopt_mnist_jigen.yaml b/examples/benchmark/benchmark_fbopt_mnist_jigen.yaml
@@ -29,17 +29,10 @@ domainlab_args:
 
 
 Shared params:
-  ini_setpoint_ratio:
-    min: 0.95
-    max: 0.99
-    num: 2
-    distribution: uniform
-
   k_i_gain:
     min: 0.0001
     max: 0.01
     num: 2
-    step: 0.0001
     distribution: uniform
 
   mu_init:

diff --git a/examples/benchmark/benchmark_fbopt_pacs_jigen.yaml b/examples/benchmark/benchmark_fbopt_pacs_jigen.yaml
@@ -39,8 +39,8 @@ Shared params:
     distribution: loguniform
 
   mu_init:
-    min: 0.00001
-    max: 0.0001
+    min: 0.000001
+    max: 0.00005
     num: 3
     distribution: loguniform
 
@@ -52,7 +52,7 @@ Shared params:
 
   gamma_reg:
     min: 0.01
-    max: 10
+    max: 10_000
     num: 10
     distribution: loguniform