File size: 637 Bytes
16eb59e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
train_loop () {
    local WANDB_TAGS="benchmark_$(git rev-parse --short HEAD) host_$(hostname)"
    local algo
    local env
    local seed
    local WANDB_PROJECT_NAME="${WANDB_PROJECT_NAME:-rl-algo-impls-benchmarks}"
    local args=()
    (( VIRTUAL_DISPLAY == 1)) && args+=("--virtual-display")
    local SEEDS="${SEEDS:-1 2 3}"
    for algo in $(echo $1); do
        for env in $(echo $2); 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 $WANDB_PROJECT_NAME ${args[@]}
            done
        done
    done
}