File size: 437 Bytes
0bbce05
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
source benchmarks/train_loop.sh

# export WANDB_PROJECT_NAME="rl-algo-impls"

BENCHMARK_MAX_PROCS="${BENCHMARK_MAX_PROCS:-5}"

ALGOS=(
    # "vpg"
    # "dqn"
    "ppo"
)
ENVS=(
    "impala-PongNoFrameskip-v4"
    "impala-BreakoutNoFrameskip-v4"
    "impala-SpaceInvadersNoFrameskip-v4"
    "impala-QbertNoFrameskip-v4"
    "impala-CarRacing-v0"
)
train_loop "${ALGOS[*]}" "${ENVS[*]}" | xargs -I CMD -P $BENCHMARK_MAX_PROCS bash -c CMD