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")