File size: 543 Bytes
e9e96b1
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
while getopts a:e:s:p: flag
do
    case "${flag}" in
        a) algo=${OPTARG};;
        e) envs=${OPTARG};;
        s) seeds=${OPTARG};;
        p) project_name=${OPTARG};;
    esac
done

WANDB_TAGS=$(bash scripts/tags_benchmark.sh)
project_name="${project_name:-rl-algo-impls-benchmarks}"
seeds="${seeds:-1 2 3}"
for env in $(echo $envs); do
    for seed in $seeds; do
        echo python train.py --algo $algo --env $env --seed $seed --pool-size 1 --wandb-tags $WANDB_TAGS --wandb-project-name $project_name --virtual-display
    done
done