diff --git a/MLPerf/training/image_segmentation/pytorch/submit_benchmark_jobs.sh b/MLPerf/training/image_segmentation/pytorch/submit_benchmark_jobs.sh
index 27c84da6f5cd07ac824e797bdc40e3750b9deea8..916799a2b7e4980bca99c5b0f126cc95b7abb227 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 fd1a27ba28dd1f5813b30e72609b63f171aad843..b7ad2773692d11c7ace525b991cfd424f5dfc44b 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 3c4f049abe6d922543c0094a9ebc64e2ee0ed9d0..2397e4f633efcb25507881860a344a048d02270d 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")