From 47b10d1b3201de9e202aaea4619549db6a0f85e1 Mon Sep 17 00:00:00 2001
From: Xuan Gu <xuan.gu@liu.se>
Date: Wed, 11 Oct 2023 16:30:19 +0000
Subject: [PATCH] Update 2 files
- /MLPerf/training/image_segmentation/pytorch/benchmark.txt
- /MLPerf/training/image_segmentation/pytorch/U-Net3D.def
---
.../image_segmentation/pytorch/U-Net3D.def | 6 +++++-
.../image_segmentation/pytorch/benchmark.txt | 14 ++++++++++++++
2 files changed, 19 insertions(+), 1 deletion(-)
create mode 100644 MLPerf/training/image_segmentation/pytorch/benchmark.txt
diff --git a/MLPerf/training/image_segmentation/pytorch/U-Net3D.def b/MLPerf/training/image_segmentation/pytorch/U-Net3D.def
index 8771fe0..7b4a50c 100644
--- a/MLPerf/training/image_segmentation/pytorch/U-Net3D.def
+++ b/MLPerf/training/image_segmentation/pytorch/U-Net3D.def
@@ -22,4 +22,8 @@ mv training/image_segmentation/pytorch/* /workspace/unet3d/
rm -rf training
pip install --upgrade pip
-pip install --disable-pip-version-check -r requirements.txt
\ No newline at end of file
+pip install --disable-pip-version-check -r requirements.txt
+
+git clone https://github.com/neheller/kits19
+cd kits19
+pip3 install -r requirements.txt
\ No newline at end of file
diff --git a/MLPerf/training/image_segmentation/pytorch/benchmark.txt b/MLPerf/training/image_segmentation/pytorch/benchmark.txt
new file mode 100644
index 0000000..716fc21
--- /dev/null
+++ b/MLPerf/training/image_segmentation/pytorch/benchmark.txt
@@ -0,0 +1,14 @@
+
+
+MODLE_NAME=U-Net3D
+# MODULE_VERSION=21.11.0
+CONTAINER_DIR=/proj/nsc_testing/xuan/containers/${MODLE_NAME}.sif
+WORK_DIR=/proj/nsc_testing/xuan/berzelius-benchmarks/MLPerf/training/image_segmentation/pytorch
+
+cd $WORK_DIR
+mkdir raw-data-dir
+cd raw-data-dir
+git clone https://github.com/neheller/kits19
+cd kits19
+pip3 install -r requirements.txt
+python3 -m starter_code.get_imaging
\ No newline at end of file
--
GitLab