diff --git a/test_tipc/prepare.sh b/test_tipc/prepare.sh
index 675ef92d9..646e7f4d8 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 cecfa93e7..39503dd00 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"