fix tipc lite_train_whole_infer bug

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

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"