diff --git a/run_model.sh b/run_model.sh
index 7559534..ea6fffb 100644
--- a/run_model.sh
+++ b/run_model.sh
@@ -15,7 +15,7 @@ EPOCHS=100
 for ((i = ${N1}; i <= ${N2}; i++))
 do
     python main.py --mode train \
-                --train ${DATADIR}/currentData.csv \
+                --train ${DATADIR} \
                 --model ${RESDIR}/${OUTPUT}_${TYPE}_${i} \
                 --epochs ${EPOCHS} \
                 --embd 128 \