diff --git a/test_tipc/prepare.sh b/test_tipc/prepare.sh
index 675ef92d973e0413b944477e9532cd3175b72bfc..646e7f4d8dfef3f1bae4806985e037f9f1f5708c 100644
--- a/test_tipc/prepare.sh
+++ b/test_tipc/prepare.sh
@@ -118,9 +118,7 @@ if [ ${MODE} = "lite_train_lite_infer" ] || [ ${MODE} = "lite_train_whole_infer"
     cd ILSVRC2012 
     mv train.txt train_list.txt
     mv val.txt val_list.txt
-    if [ ${MODE} = "lite_train_lite_infer" ];then
-	cp -r train/* val/
-    fi
+    cp -r train/* val/
     cd ../../
 elif [ ${MODE} = "whole_infer" ] || [ ${MODE} = "klquant_whole_infer" ];then
     # download data
diff --git a/test_tipc/test_train_inference_python.sh b/test_tipc/test_train_inference_python.sh
index cecfa93e70d4767b43f667ab01134d03409458a0..39503dd004a1f1811cd5dca1ab27adbf1bb51afa 100644
--- a/test_tipc/test_train_inference_python.sh
+++ b/test_tipc/test_train_inference_python.sh
@@ -87,7 +87,9 @@ benchmark_key=$(func_parser_key "${lines[49]}")
 benchmark_value=$(func_parser_value "${lines[49]}")
 infer_key1=$(func_parser_key "${lines[50]}")
 infer_value1=$(func_parser_value "${lines[50]}")
-
+if [ ! $epoch_num ]; then
+  epoch_num=2
+fi
 LOG_PATH="./test_tipc/output"
 mkdir -p ${LOG_PATH}
 status_log="${LOG_PATH}/results_python.log"