update config

aai-institute · Nov 10, 2024 · 4d84c7f · 4d84c7f
1 parent a07bd78
commit 4d84c7f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/experiments/cfair/cfair.yaml b/experiments/cfair/cfair.yaml
@@ -11,7 +11,7 @@ experiments:
       max_t: 1000000
       grace_period: 1000000
       reduction_factor: 2
-    num_hyperopt_samples: &num_hyperopt_samples 1
+    num_hyperopt_samples: &num_hyperopt_samples 4
     gpus_per_trial: &gpus_per_trial 1
     cpus_per_trial: &cpus_per_trial 1
     tuner_params: &tuner_params
@@ -62,7 +62,7 @@ experiments:
             norm_distribution:
               __object__: pyro.distributions.LogNormal
               loc: 
-                __eval__: torch.zeros(1).to("cuda")
+                __eval__: torch.zeros(7).to("cuda")
               scale: 
                 __eval__: (.5 * torch.ones(1)).to("cuda")
           use_lu: true