From 112f668d82dd697b37a0dcacedc8da7915bc3088 Mon Sep 17 00:00:00 2001 From: Xuan Gu <xuan.gu@liu.se> Date: Fri, 13 Oct 2023 17:54:21 +0000 Subject: [PATCH] Update file submit_benchmark_jobs.sh --- .../PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh b/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh index 2397e4f..8967862 100644 --- a/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh +++ b/NVIDIA/DeepLearningExamples/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh @@ -2,13 +2,13 @@ set -e export MODEL_NAME=nnunet_for_pytorch -export MODLE_VERSION=21.11.0 +export MODLE_VERSION=latest 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 +mkdir -p $WORK_DIR/sbatch_out $WORK_DIR/sbatch_scripts $WORK_DIR/results benchmark_modes=("train" "predict") # node_types=("thin" "fat") node_types=("thin") -- GitLab