From aa1dbb4ea509ad6bde6fb24d5ab1e70b81808868 Mon Sep 17 00:00:00 2001
From: Xuan Gu <xuan.gu@liu.se>
Date: Thu, 28 Sep 2023 16:25:38 +0000
Subject: [PATCH] Update file submit_benchmark_jobs.sh
---
PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh b/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh
index 7f66778..424430d 100644
--- a/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh
+++ b/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh
@@ -15,7 +15,7 @@ for nodes in {1..1}; do
if [ "${node_type}" = "thin" ]; then
batch_size=512
else
- batch_size=2048
+ batch_size=1024
fi
echo dim ${dim}, nodes ${nodes}, gpus ${gpus}, batch_size ${batch_size}, benchmark_mode ${benchmark_mode}, node_type ${node_type}
--
GitLab