diff --git a/MLPerf/training/image_segmentation/pytorch/submit_benchmark_jobs.sh b/MLPerf/training/image_segmentation/pytorch/submit_benchmark_jobs.sh
index 916799a2b7e4980bca99c5b0f126cc95b7abb227..c8b17ae37ffd042aff8c889eea2b970b6c7a797c 100644
--- a/MLPerf/training/image_segmentation/pytorch/submit_benchmark_jobs.sh
+++ b/MLPerf/training/image_segmentation/pytorch/submit_benchmark_jobs.sh
@@ -20,9 +20,9 @@ for nodes in {1..1}; do
         
 
                     if [ "${node_type}" = "thin" ]; then
-                        batch_size=512
+                        batch_size=2
                     else
-                        batch_size=1024
+                        batch_size=2
                     fi
 
                     echo dim ${dim}, nodes ${nodes}, gpus ${gpus}, batch_size ${batch_size}, benchmark_mode ${benchmark_mode}, node_type ${node_type}