File size: 547 Bytes
3d6ce6f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
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
} |