A2C playing PongNoFrameskip-v4 from https://github.com/sgoodfriend/rl-algo-impls/tree/0760ef7d52b17f30219a27c18ba52c8895025ae3
1413f39
| source benchmarks/train_loop.sh | |
| # export WANDB_PROJECT_NAME="rl-algo-impls" | |
| BENCHMARK_MAX_PROCS="${BENCHMARK_MAX_PROCS:-6}" | |
| ALGOS=( | |
| # "vpg" | |
| # "dqn" | |
| # "ppo" | |
| "a2c" | |
| ) | |
| ENVS=( | |
| # Basic | |
| "CartPole-v1" | |
| "MountainCar-v0" | |
| "MountainCarContinuous-v0" | |
| "Acrobot-v1" | |
| "LunarLander-v2" | |
| "BipedalWalker-v3" | |
| # 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 | |