File size: 694 Bytes
e013d6a |
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"
"Walker2DBulletEnv-v0"
"HopperBulletEnv-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
|