From c334cd18fce53b1bd37b707de75d22687b1c0a2e Mon Sep 17 00:00:00 2001
From: "michael.divia" <michael.divia@etu.hesge.ch>
Date: Wed, 9 Apr 2025 15:08:42 +0200
Subject: [PATCH] Baobab good this time
---
Python/pokedex_ResNet50.py | 2 +-
slurm/train_ResNet50.sh | 8 ++++----
2 files changed, 5 insertions(+), 5 deletions(-)
diff --git a/Python/pokedex_ResNet50.py b/Python/pokedex_ResNet50.py
index 171fcc1..dc528a7 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 f545644..1dbf762 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
--
GitLab