#!/bin/bash
#SBATCH -t 03:0git stat0:00
#SBATCH -p gpu
#SBATCH --gpus=1
#SBATCH --ntasks 1
#SBATCH --cpus-per-task=18
#SBATCH --mem=40G

module load 2022
module load Python/3.10.4-GCCcore-11.3.0
module load TensorFlow/2.11.0-foss-2022a-CUDA-11.7.0
module load h5py/3.7.0-foss-2022a

echo "executing python train.py"
python train.py train
