module purge

# modules load

module load openmpi/4.0.2-cuda
module load tensorflow-gpu/py3/2.1.0+hvd-0.19
module load git

GIT_ROOT=`git rev-parse --show-toplevel`
DG_DIR=$GIT_ROOT/Deepgalaxy/
DG_TRAIN=$DG_DIR/DeepGalaxy-master/dg_train.py