#!/bin/bash -l
#SBATCH -J DeepGalaxyTrain
#SBATCH -o DeepGalaxyTrain_out.txt
#SBATCH -e DeepGalaxyTrain_err.txt

#SBATCH -t 1:00:00

#SBATCH --partition gpu_titanrtx

#SBATCH --nodes 8
#SBATCH --ntasks-per-node=4
#SBATCH --gres=gpu:4



module purge
module load 2020
module load TensorFlow/2.1.0-foss-2019b-Python-3.7.4-CUDA-10.1.243
# module load TensorFlow/1.15.0-foss-2019b-Python-3.7.4-10.1.243

mpirun -np 32 python dg_train.py -f output_bw_512.hdf5 --arch EfficientNetB4 --epochs 10 --noise 0.3 --batch-size 4