PPO playing impala-PongNoFrameskip-v4 from https://github.com/sgoodfriend/rl-algo-impls/tree/e47a44c4d891f48885af0b1605b30d19fc67b5af
be9c115
| 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 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 | |
| done | |
| done | |
| done | |
| } |