#!/bin/bash

#SBATCH -p GPU-shared

#SBATCH -t 48:00:00

#SBATCH --gres=gpu:p100:1

#SBATCH --no-requeue



# Usage: ./train.sh config batch_size



module load python/3.6.4_gcc5_np1.14.5

module load cuda/9.0



cd $SCRATCH/yolact



python3 train.py --config $1 --batch_size $2 --save_interval 5000 &>logs/$1_log