From e59438a9029e2679e563c80fa90971b2bd4bdcab Mon Sep 17 00:00:00 2001 From: Xuan Gu <xuan.gu@liu.se> Date: Thu, 12 Oct 2023 22:56:05 +0000 Subject: [PATCH] Update file submit_benchmark_jobs.sh --- .../image_segmentation/pytorch/submit_benchmark_jobs.sh | 2 +- 1 file changed, 1 insertion(+), 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 12c3c5a..27c84da 100644 --- a/MLPerf/training/image_segmentation/pytorch/submit_benchmark_jobs.sh +++ b/MLPerf/training/image_segmentation/pytorch/submit_benchmark_jobs.sh @@ -4,7 +4,7 @@ set -e 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=nodeimage +export GPU_RESERVATION= mkdir -p $WORK_DIR/sbatch_out $WORK_DIR/sbatch_scripts $WORK_DIR/results benchmark_modes=("train" "predict") -- GitLab