-
Notifications
You must be signed in to change notification settings - Fork 0
/
run_hpo.sh
executable file
·34 lines (31 loc) · 1.1 KB
/
run_hpo.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
#!/bin/bash
optimisers=(
'SGD'
'SGDmwd'
'Adam'
'KFAC'
'AdamQLR_Damped'
)
datasets=(
'uci_energy'
'uci_protein'
'fashion_mnist'
'svhn'
'cifar-10'
# 'penn_treebank_gpt2_reset'
)
for dataset in ${datasets[@]}
do
for optimiser in ${optimisers[@]}
do
echo "================================================================================"
echo "================================================================================"
echo "================================================================================"
echo "Starting ${dataset}, ${optimiser}..."
XLA_PYTHON_CLIENT_PREALLOCATE=false python hyperparameter_optimisation.py -c "configs/${dataset}.yaml" "configs/${optimiser}.yaml" configs/ASHA.yaml
echo "Finished ${dataset}, ${optimiser}."
done
done
echo "================================================================================"
echo "================================================================================"
echo "================================================================================"