Submit_HPC_train_regression.sh

#!/bin/bash

#SBATCH -p gpu
#SBATCH -t 5-00:00:00
#SBATCH -n 1
#SBATCH -c 16
#SBATCH --gpus-per-node=1
#SBATCH -J GenNet_regression
#SBATCH --mem=100G
#SBAYCH --mem-per-gpu=39G
#SBATCH -o /home/ahilten/repositories/GenNet/GenNet_utils/SLURM_logs/out_%j.log
#SBATCH -e /home/ahilten/repositories/GenNet/GenNet_utils/SLURM_logs/error_%j.log

# Load the modules


module purge
module load 2021
module load CUDA/11.3.1
module load cuDNN/8.2.1.32-CUDA-11.3.1


source $HOME/env_GenNet/bin/activate

cd /home/ahilten/repositories/GenNet/

python GenNet.py train /home/ahilten/repositories/pheno_height/Input_files/ $1 -genotype_path /projects/0/emc17610/nvidia/UKBB_HRC_imputed/genotype/ -problem_type regression -lr $2 -bs $3 -L1 $4 -network_name regression_height -epoch_size=50000 -patience=50