Update file submit_benchmark_sbatch.sh

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}