train_HGD_positive_COM_best_split.sbatch

#!/bin/bash --login
#SBATCH --time=24:00:00
#SBATCH --nodes=1
#SBATCH --gpus-per-node=v100:1
#SBATCH --cpus-per-gpu=6 
#SBATCH --mem=32G 
#SBATCH --constraint=intel
#SBATCH --partition=batch
#SBATCH --mail-type=ALL
#SBATCH --output=./HGD_positive_COM_best_split_logs/%x-%j-slurm.out
#SBATCH --error=./HGD_positive_COM_best_split_logs/%x-%j-slurm.err

set -e
module purge
module load cuda/11.2.2
module load cudnn/8.1.1-cuda11.2.2 

# Load software stack
conda activate ./env

python prime_NAS_HGD_positive_COM_best_split.py