From d1af2d4a9deb8c80a9aabc563db5ddafae48c64e Mon Sep 17 00:00:00 2001
From: Xuan Gu <xuan.gu@liu.se>
Date: Wed, 11 Oct 2023 15:39:57 +0000
Subject: [PATCH] Update 2 files

- /NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh
- /NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/generate_benchmark_jobs.sh
---
 .../PyTorch/Segmentation/nnUNet/generate_benchmark_jobs.sh      | 2 +-
 .../PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh        | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/generate_benchmark_jobs.sh b/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/generate_benchmark_jobs.sh
index 5a04b7a..fd1a27b 100644
--- a/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/generate_benchmark_jobs.sh
+++ b/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/generate_benchmark_jobs.sh
@@ -18,7 +18,7 @@ EOT
 if [ "${6}" = "thin" ]; then
     cat <<EOT >>  $SBATCH_DIR
 #SBATCH -C "thin"    
-#SBATCH --reservation=nodeimage
+#SBATCH --reservation=$GPU_RESERVATION
 EOT
 else
     cat <<EOT >>  $SBATCH_DIR
diff --git a/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh b/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh
index 729eba6..0d1ef6f 100644
--- a/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh
+++ b/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh
@@ -5,6 +5,7 @@ export MODULE_NAME=nnunet_for_pytorch
 export MODULE_VERSION=21.11.0
 export WORK_DIR=/proj/nsc_testing/xuan/berzelius-benchmarks/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet
 export CONTAINER_DIR=/proj/nsc_testing/xuan/containers/${MODULE_NAME}_${MODULE_VERSION}.sif
+export GPU_RESERVATION=nodeimage
 mkdir -p $WORK_DIR/sbatch_out $WORK_DIR/sbatch_scripts $WORK_DIR/results
 
 benchmark_modes=("train" "predict")
-- 
GitLab