From f6f2a20126a3a9c6b348c5be0ae6ab1fcdcaec53 Mon Sep 17 00:00:00 2001
From: dongshuilong <dongshuilong@baidu.com>
Date: Fri, 24 Dec 2021 03:11:26 +0000
Subject: [PATCH] fix tipc lite_train_whole_infer bug

---
 test_tipc/prepare.sh                     | 4 +---
 test_tipc/test_train_inference_python.sh | 4 +++-
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/test_tipc/prepare.sh b/test_tipc/prepare.sh
index 675ef92d..646e7f4d 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 cecfa93e..39503dd0 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"
-- 
GitLab