File size: 922 Bytes
cce40ac |
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 |