#!/bin/bash

#SBATCH --job-name=py-matinv     # create a short name for your job

#SBATCH --nodes=1                # node count

#SBATCH --ntasks=1               # total number of tasks across all nodes

#SBATCH --cpus-per-task=1        # cpu-cores per task (>1 if multi-threaded tasks)

#SBATCH --mem-per-cpu=50G         # memory per cpu-core (4G is default)

#SBATCH --gres=gpu:1             # number of gpus per node

#SBATCH --time=40:00:00          # total run time limit (HH:MM:SS)

#SBATCH --mail-type=begin        # send email when job begins

#SBATCH --mail-type=end          # send email when job ends

#SBATCH --mail-type=fail         # send email if job fails

#SBATCH --mail-user=eshen@princeton.edu

#SBATCH --constraint=gpu80

module purge

module load anaconda3/2024.2

conda activate tf2-gpu

conda activate ed2

pip install -r requirements.txt



python train.py --config train.json