Newer
Older
#!/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