train_Xception.sh

#!/bin/sh
#SBATCH --job-name=ResNet50
#SBATCH --output=Xception_%j.out
#SBATCH --partition=shared-gpu
#SBATCH --gres=gpu:1,VramPerGpu:80G
#SBATCH --cpus-per-task=2
#SBATCH --mem=16G
#SBATCH --time=05:00:00
#SBATCH --mail-type=FAIL

# Load modules
module purge
module load GCC/11.3.0  OpenMPI/4.1.4 TensorFlow/2.11.0-CUDA-11.7.0
module load cuDNN/8.4.1.50-CUDA-11.7.0
module load scikit-learn/1.1.2

# Run your script
srun python ../python/pokedex_Xception.py --hpc yes