diff --git a/run_model.sh b/run_model.sh
index 45fce6b..7559534 100644
--- a/run_model.sh
+++ b/run_model.sh
@@ -1,28 +1,28 @@
 #!/usr/bin/env bash
 
 
-N=$1
-OUTPUT=$2
-DEPTH=$3
-TYPE=$4
-RESDIR=$5
+N1=$1
+N2=$2
+OUTPUT=$3
+DEPTH=$4
+TYPE=$5
+RESDIR=$6
 mkdir -p /tmp/rk/${RESDIR}
-DATADIR=$6
+DATADIR=$7
 
 EPOCHS=100
 
-for i in {1..$N}
+for ((i = ${N1}; i <= ${N2}; i++))
 do
     python main.py --mode train \
                 --train ${DATADIR}/currentData.csv \
-                --model ${RESDIR}/${OUTPUT}_${TYPE}_$i \
-                --epochs $EPOCHS \
+                --model ${RESDIR}/${OUTPUT}_${TYPE}_${i} \
+                --epochs ${EPOCHS} \
                 --embd 128 \
                 --filter_embd 256 --kernel_embd 8 --dense_embd 128 \
                 --domain_embd 32 \
                 --filter_main 32  --kernel_main 8 --dense_main 1024 \
                 --batch 128 \
-#                --balanced_weights \
                 --model_output ${OUTPUT} \
                 --type ${TYPE} \
                 --depth ${DEPTH}