#!/bin/bash

for seed in 123 231 321
do
    echo "Running experiment with seed $seed"
    python3 baselines/ppo/ppo_memtasks.py \
        --env_id=RememberShapeAndColor3x3-v0 \
        --exp-name=remember-shape-and-color-3x3-v0 \
        --capture-video \
        --save-model \
        --track \
        --num-steps=60 \
        --num_eval_steps=180 \
        --include-state \
        --seed=$seed \
        --total-timesteps=5_000_000 \
        --no-finite-horizon-gae \
        --eval-freq=25 \
        --gae-lambda=0.9 \
        --gamma=0.8 \
        --update-epochs=4
done