diff --git a/PyTorch/Segmentation/nnUNet/submit_benchmark_sbatch.sh b/PyTorch/Segmentation/nnUNet/submit_benchmark_sbatch.sh
index cafc395ed16b5b2f3a45b6f6b704b4dd15116f46..2e5f89a951c9d423a43e84f820fe0a6168fbe35e 100644
--- a/PyTorch/Segmentation/nnUNet/submit_benchmark_sbatch.sh
+++ b/PyTorch/Segmentation/nnUNet/submit_benchmark_sbatch.sh
@@ -6,7 +6,7 @@ WORK_DIR=/proj/nsc_testing/xuan/berzelius-benchmarks/PyTorch/Segmentation/nnUNet
 dim=2
 for nodes in {1..1}; do
     for gpus in {1,8}; do
-        for batch_size in 128; do
+        for batch_size in 256; do
             for iteration in {1..1}; do
 
                 echo dim ${dim}, nodes ${nodes}, gpus ${gpus}, batch_size ${batch_size}, iteration ${iteration}