diff --git a/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh b/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh
index ee4a01efb219771c7ab53f56e92fc2b962002ab1..913cafc40aca86321d6c102b4d96446c0785ba8b 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=256
                     else
-                        batch_size=512
+                        batch_size=1024
                     fi
 
                     echo dim ${dim}, nodes ${nodes}, gpus ${gpus}, batch_size ${batch_size}, benchmark_mode ${benchmark_mode}, node_type ${node_type}