diff --git a/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh b/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh
index 7f667782638405a2383ceb2255cfa4c32742a1a7..424430dac8d721b7480586e73e2eab3ae4363c47 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}