#!/bin/bash

#SBATCH --mail-user=kaiyuekarinwu@gmail.com
#SBATCH --mail-type=ALL
#SBATCH --ntasks=256
#SBATCH --nodes=2
#SBATCH --mem-per-cpu=10G
#SBATCH --partition=k2-himem
#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
export OMP_NUM_THREADS=1

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

