diff --git a/test/test.sh b/test/test.sh
index cd4d93d91eab4264f41ea8fc13ca4e5bcd57d5d2..b43b1a615723653a3095bd4b1feb6bb2a5476e60 100644
--- a/test/test.sh
+++ b/test/test.sh
@@ -116,8 +116,12 @@ for train_model in ${train_model_list[*]}; do
                 if [ "${model_name}" = "det" ]; then 
                     export rec_batch_size_list=( "1" )
                     inference="tools/infer/predict_det.py"
+                    det_model_dir=${save_log}/export_inference/
+                    rec_model_dir=""
                 elif [ "${model_name}" = "rec" ]; then
                     inference="tools/infer/predict_rec.py"
+                    rec_model_dir=${save_log}/export_inference/
+                    det_model_dir=""
                 fi
                 # inference 
                 for device in ${devices[*]}; do