# Evaluate, choose an appropriate epoch
CUDA_VISIBLE_DEVICES=0 python3 eval_new.py --env darkroom_heldout \
    --envs 100000 \
    --H 50 \
    --dim 4 \
    --lr 0.001 \
    --layer 4 \
    --head 4 \
    --shuffle \
    --epoch 700 \
    --seed 1


# Evaluate, choose an appropriate epoch
CUDA_VISIBLE_DEVICES=0 python3 eval_new.py --env darkroom_heldout_lookahead_pred_reward \
    --envs 100000 \
    --H 50 \
    --dim 4 \
    --lr 0.001 \
    --layer 4 \
    --head 4 \
    --shuffle \
    --epoch 700 \
    --seed 1