DQN playing PongNoFrameskip-v4 from https://github.com/sgoodfriend/rl-algo-impls/tree/e8bc541d8b5e67bb4d3f2075282463fb61f5f2c6
c34fd6e
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 | |
} |