From 4dd40523557c8f704a6dbe810dec354f6aab39b8 Mon Sep 17 00:00:00 2001
From: Xuan Gu <xuan.gu@liu.se>
Date: Thu, 12 Oct 2023 23:04:19 +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 c8b17ae..c837ccd 100644
--- a/MLPerf/training/image_segmentation/pytorch/submit_benchmark_jobs.sh
+++ b/MLPerf/training/image_segmentation/pytorch/submit_benchmark_jobs.sh
@@ -1,7 +1,7 @@
#!/bin/bash
set -e
-export MODEL_NAME=nnunet_for_pytorch
+export MODEL_NAME=U-Net3D
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=
--
GitLab