File tree Expand file tree Collapse file tree 1 file changed +3
-1
lines changed
Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Original file line number Diff line number Diff line change @@ -10,18 +10,20 @@ methods=("bo" "ibnn" "turbo" "dycors" "snbo")
1010dims=(10 25 50)
1111max_evals=(500 1000 1000)
1212problems=(" ackley" " rastrigin" " levy" )
13+ neurons=(128 256 256)
1314
1415for seed in $( seq 1 10) ; do
1516 for dim_idx in " ${! dims[@]} " ; do
1617
1718 dim=${dims[$dim_idx]}
1819 n_init=$(( 2 * $dim ))
1920 max_eval=${max_evals[$dim_idx]}
21+ neuron=${neurons[$dim_idx]}
2022
2123 for prob_idx in " ${! problems[@]} " ; do
2224 for method_idx in " ${! methods[@]} " ; do
2325 python algorithms/optimize.py --method ${methods[$method_idx]} --problem ${problems[$prob_idx]} --dim $dim --n_init $n_init \
24- --max_evals $max_eval --seed $seed --neurons 128 128 --act_funcs " GELU" " GELU"
26+ --max_evals $max_eval --seed $seed --neurons $neuron $neuron --act_funcs " GELU" " GELU"
2527 done
2628 done
2729
You can’t perform that action at this time.
0 commit comments