File size: 710 Bytes
6d1ad4f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
33
34
source benchmarks/train_loop.sh

# export WANDB_PROJECT_NAME="rl-algo-impls"
export VIRTUAL_DISPLAY=1

BENCHMARK_MAX_PROCS="${BENCHMARK_MAX_PROCS:-6}"

ALGOS=(
    # "vpg"
    "dqn"
    # "ppo"
)
ENVS=(
    # Basic
    "CartPole-v1"
    "MountainCar-v0"
    # "MountainCarContinuous-v0"
    "Acrobot-v1"
    "LunarLander-v2"
    # # PyBullet
    # "HalfCheetahBulletEnv-v0"
    # "AntBulletEnv-v0"
    # "HopperBulletEnv-v0"
    # "Walker2DBulletEnv-v0"
    # # CarRacing
    # "CarRacing-v0"
    # Atari
    "PongNoFrameskip-v4"
    "BreakoutNoFrameskip-v4"
    "SpaceInvadersNoFrameskip-v4"
    "QbertNoFrameskip-v4"
)
train_loop "${ALGOS[*]}" "${ENVS[*]}" | xargs -I CMD -P $BENCHMARK_MAX_PROCS bash -c CMD