diff --git a/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh b/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh
index 6e49a66456c00b32c616a7ce4a26da1821c9e012..5c21299254ce2ddb1df7e6539bb138a1f3e9a98e 100644
--- a/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh
+++ b/PyTorch/Segmentation/nnUNet/submit_benchmark_jobs.sh
@@ -8,19 +8,23 @@ node_types=("thin" "fat")
 dim=2
 for nodes in {1..1}; do
     for gpus in {1,8}; do
-        for batch_size in 256; do
-            for benchmark_mode in "${benchmark_modes[@]}"; do
-                for node_type in "${node_types[@]}"; do
+        for benchmark_mode in "${benchmark_modes[@]}"; do
+            for node_type in "${node_types[@]}"; do
         
 
+                    if [ "${node_type}" = "thin" ]; then
+                        batch_size = 256
+                    else
+                        batch_size = 512
+                    fi
+
                     echo dim ${dim}, nodes ${nodes}, gpus ${gpus}, batch_size ${batch_size}, benchmark_mode ${benchmark_mode}, node_type ${node_type}
 
                     # For single node
                     bash $WORK_DIR/generate_benchmark_jobs.sh ${dim} ${nodes} ${gpus} ${batch_size} ${benchmark_mode} ${node_type}
                     SBATCH_DIR=$WORK_DIR/sbatch_scripts/benchmark_${node_type}_${benchmark_mode}_dim${dim}_nodes${nodes}_gpus${gpus}_batchsize${batch_size}.sbatch
                     #sbatch $SBATCH_DIR
-                    sleep 1 
-                done
+                    #sleep 1 
             done
         done
     done