From adafd6f6aee8ca24d51774df87c487aadf75b643 Mon Sep 17 00:00:00 2001
From: Xuan Gu <xuan.gu@liu.se>
Date: Thu, 12 Oct 2023 22:59:29 +0000
Subject: [PATCH] Update 3 files
- /NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/generate_benchmark_jobs.sh
- /NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh
- /MLPerf/training/image_segmentation/pytorch/submit_benchmark_jobs.sh
---
.../image_segmentation/pytorch/submit_benchmark_jobs.sh | 1 +
.../PyTorch/Segmentation/nnUNet/generate_benchmark_jobs.sh | 2 +-
.../PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh | 1 +
3 files changed, 3 insertions(+), 1 deletion(-)
diff --git a/MLPerf/training/image_segmentation/pytorch/submit_benchmark_jobs.sh b/MLPerf/training/image_segmentation/pytorch/submit_benchmark_jobs.sh
index 27c84da..916799a 100644
--- a/MLPerf/training/image_segmentation/pytorch/submit_benchmark_jobs.sh
+++ b/MLPerf/training/image_segmentation/pytorch/submit_benchmark_jobs.sh
@@ -5,6 +5,7 @@ export MODEL_NAME=nnunet_for_pytorch
export WORK_DIR=/proj/nsc_testing/xuan/berzelius-benchmarks/MLPerf/training/image_segmentation/pytorch
export CONTAINER_DIR=/proj/nsc_testing/xuan/containers/${MODEL_NAME}.sif
export GPU_RESERVATION=
+export TIME__RESERVATION=0-0:20:00
mkdir -p $WORK_DIR/sbatch_out $WORK_DIR/sbatch_scripts $WORK_DIR/results
benchmark_modes=("train" "predict")
diff --git a/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/generate_benchmark_jobs.sh b/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/generate_benchmark_jobs.sh
index fd1a27b..b7ad277 100644
--- a/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/generate_benchmark_jobs.sh
+++ b/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/generate_benchmark_jobs.sh
@@ -10,7 +10,7 @@ cat <<EOT > $SBATCH_DIR
#SBATCH -A nsc
#SBATCH --nodes=${2}
#SBATCH --gpus=${3}
-#SBATCH --time=0-0:20:00
+#SBATCH --time=$TIME_RESERVATION
#SBATCH --output=$SBATCH_OUT_DIR
EOT
diff --git a/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh b/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh
index 3c4f049..2397e4f 100644
--- a/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh
+++ b/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh
@@ -6,6 +6,7 @@ export MODLE_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/${MODEL_NAME}_${MODLE_VERSION}.sif
export GPU_RESERVATION=nodeimage
+export TIME_RESERVATIOn=0-0:20:00
mkdir -p $WORK_DIR/sbatch_out $WORK_DIR/sbatch_scripts $WORK_DIR/results
benchmark_modes=("train" "predict")
--
GitLab