train2_20.job

#!/bin/bash
#SBATCH --job-name=20FinchNetTPFP5Fold
#SBATCH --time=0-10:00:00
#SBATCH --ntasks=1

### just increase this as necessary
#SBATCH --mem=32g
#SBATCH --cpus-per-task=1
#SBATCH --gres=gpu:1
#SBATCH --account=mb20
#SBATCH --partition=m3g,m3h

module load pytorch/1.3-cuda10

python train2_20.py