#!/bin/bash

#SBATCH --mail-user=kaiyuekarinwu@gmail.com
#SBATCH --mail-type=ALL
#SBATCH --partition=k2-epsrc-gpu
#SBATCH --gres gpu:v100:2
#SBATCH --output=/users/kwu/sharedscratch/output/%j.out
#SBATCH --error=/users/kwu/sharedscratch/error/%j.err

module load apps/anaconda3/2022.10/bin
source ~/.bashrc
conda activate py36-2

OPENAI_LOG_FORMAT=tensorboard,stdout,csv,log mpirun -np 2 python -m baselines.run_exp2 --env=FishingDerbyNoFrameskip-v4 --num_timesteps=1e10 --save_path=./models/test2 --log_path=./logs/test2 --num_env=1

