#!/bin/bash

# SBATCH parameters here

conda activate radt

### FetchReachObstacle ###

python3 train.py \
        --project "baselines_radt" \
        --env "FetchReachObstacle" \
        --method "AMlag" \ 
        --expert_percent=0 \
        --random_percent=1 \
        --massless=1 \
        --n-epochs=500 \
        --n-test-rollouts=60 \
        --bsa_box_size=0.1


### MazeObstacle ###

# python3 train.py \
#         --project "baselines_radt" \
#         --env "PointMazeObstacle" \
#         --run_suffix "umaze300" \
#         --method "AMlag" \
#         --expert_percent=0 \
#         --random_percent=1 \
#         --maze "U_MAZE" \
#         --num_avoid=3 \
#         --n-test-rollouts=200 \
#         --n-epochs=500 \
#         --max_episode_steps=300


