File size: 922 Bytes
7f09aac
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
while getopts a:e:j:n:s:i: flag
do
    case "${flag}" in
        a) algo=${OPTARG};;
        e) env=${OPTARG};;
        j) n_jobs=${OPTARG};;
        n) study_name=${OPTARG};;
        s) seeds=${OPTARG};;
        i) increment=${OPTARG};;
    esac
done

TZ="America/Los_Angeles"
NOW=$(date +"%Y-%m-%dT%H:%M:%S")
study_name="${study_name:-$algo-$env-$NOW}"
STORAGE_PATH="sqlite:///runs/tuning.db"
increment="${increment:-100}"

mkdir -p runs
optuna create-study --study-name $study_name --storage $STORAGE_PATH --direction maximize --skip-if-exists

optimize () {
    for ((j=$increment;j<=n_jobs*100+$increment;j+=100)); do
        seed=()
        for ((s=0;s<seeds;s++)); do
            seed+="$((j+s*100/seeds)) "
        done
        echo python optimize.py --algo $algo --env $env --seed $seed --load-study --study-name $study_name --storage-path $STORAGE_PATH
    done
}

optimize | xargs -I CMD -P $n_jobs bash -c CMD