diff --git a/Python/pokedex_ResNet50.py b/Python/pokedex_ResNet50.py index 171fcc1c634678d0a46be00dddbf0a4ffe10fa2e..dc528a77540d91eaffbc0afe391de81c1ff935b3 100644 --- a/Python/pokedex_ResNet50.py +++ b/Python/pokedex_ResNet50.py @@ -17,7 +17,7 @@ print("Number of GPUs:", strategy.num_replicas_in_sync) # --- Paths --- #dataset_path = "/home/padi/Git/pokedex/Combined_Dataset" #model_output_path = "/home/padi/Git/pokedex/ResNet50" -dataset_path = "/home/users/d/divia/pokedex/Combined_Dataset" +dataset_path = "/home/users/d/divia/scratch/Combined_Dataset" model_output_path = "/home/users/d/divia/pokedex/ResNet50" os.makedirs(model_output_path, exist_ok=True) diff --git a/slurm/train_ResNet50.sh b/slurm/train_ResNet50.sh index f545644d058abf1d2738d181933c7703c73c06ea..1dbf7622a315c474a3caf3188503c96ee7dfff6f 100644 --- a/slurm/train_ResNet50.sh +++ b/slurm/train_ResNet50.sh @@ -1,11 +1,11 @@ -#!/bin/sh -#SBATCH --job-name=keras_cnn_resnet50 +l#!/bin/sh +#SBATCH --job-name=ResNet50 #SBATCH --output=ResNet50_%j.out #SBATCH --partition=shared-gpu -#SBATCH --gres=gpu:2,VramPerGpu:40G +#SBATCH --gres=gpu:1,VramPerGpu:80G #SBATCH --cpus-per-task=2 #SBATCH --mem=16G -#SBATCH --time=01:00:00 +#SBATCH --time=05:00:00 #SBATCH --mail-type=FAIL # Load modules