From 85db3bef846603564626ed0878b1d7de402adde9 Mon Sep 17 00:00:00 2001
From: Xuan Gu <xuan.gu@liu.se>
Date: Thu, 12 Oct 2023 23:00:54 +0000
Subject: [PATCH] Update file submit_benchmark_jobs.sh
---
.../image_segmentation/pytorch/submit_benchmark_jobs.sh | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/MLPerf/training/image_segmentation/pytorch/submit_benchmark_jobs.sh b/MLPerf/training/image_segmentation/pytorch/submit_benchmark_jobs.sh
index 916799a..c8b17ae 100644
--- a/MLPerf/training/image_segmentation/pytorch/submit_benchmark_jobs.sh
+++ b/MLPerf/training/image_segmentation/pytorch/submit_benchmark_jobs.sh
@@ -20,9 +20,9 @@ for nodes in {1..1}; do
if [ "${node_type}" = "thin" ]; then
- batch_size=512
+ batch_size=2
else
- batch_size=1024
+ batch_size=2
fi
echo dim ${dim}, nodes ${nodes}, gpus ${gpus}, batch_size ${batch_size}, benchmark_mode ${benchmark_mode}, node_type ${node_type}
--
GitLab