#!/bin/bash
python3 run.py distral_sac --gpu 0 --env ReacherMultistage-v0 \
    --n_policies 3 --partition task_id \
    --n_epochs 1000 --batch_size 270 \
    --num_evaluation_episodes 15 \
    --goal_type 3 \
    --vis_freq 20 --vis_num 3 --vis_height 200 --vis_width 200 --vis_fps 20 \
    --hidden_sizes 256 256 --gradient_steps_per_itr 500 \
    --min_buffer_size 1000 --target_update_tau 0.005 \
    --discount 0.99 --buffer_batch_size 384 \
    --reward_scale 1.0 --steps_per_epoch 1 \
    "$@"
