diff --git a/scripts/benchmark_sbatch_submit.sh b/scripts/benchmark_sbatch_submit.sh
index cb652f8f7ddeb7ae273e3f6710acf4212557c6d1..c142930035cb944d809784a6f7dba0238b178597 100644
--- a/scripts/benchmark_sbatch_submit.sh
+++ b/scripts/benchmark_sbatch_submit.sh
@@ -1,10 +1,8 @@
 #!/bin/bash
 
 NUM_NODES=1
-NUM_GPUS=$3
-NUM_ITERATIONS=$4
 
-dim=$1
+dim=2
 batch_size=$5
 if [ $NUM_NODES -eq 1 ]; then
     for nodes in {1..1}; do