From 343d7400f197586504eac5bd76adbad99aeb68a2 Mon Sep 17 00:00:00 2001
From: LDOUBLEV <liuvv0203@outlook.com>
Date: Fri, 19 Nov 2021 10:39:38 +0800
Subject: [PATCH] opt kl quant

---
 test_tipc/test_train_inference_python.sh | 52 ++++++++++++------------
 1 file changed, 27 insertions(+), 25 deletions(-)

diff --git a/test_tipc/test_train_inference_python.sh b/test_tipc/test_train_inference_python.sh
index b132d919..321eeb55 100644
--- a/test_tipc/test_train_inference_python.sh
+++ b/test_tipc/test_train_inference_python.sh
@@ -90,34 +90,36 @@ infer_value1=$(func_parser_value "${lines[50]}")
 
 # parser klquant_infer
 if [ ${MODE} = "klquant_whole_infer" ]; then
-    dataline=$(awk 'NR==52 NR==68{print}'  $FILENAME)
+    dataline=$(awk 'NR==1 NR==17{print}'  $FILENAME)
     lines=(${dataline})
+    model_name=$(func_parser_value "${lines[1]}")
+    python=$(func_parser_value "${lines[2]}")
     # parser inference model 
-    infer_model_dir_list=$(func_parser_value "${lines[1]}")
-    infer_export_list=$(func_parser_value "${lines[2]}")
-    infer_is_quant=$(func_parser_value "${lines[3]}")
+    infer_model_dir_list=$(func_parser_value "${lines[3]}")
+    infer_export_list=$(func_parser_value "${lines[4]}")
+    infer_is_quant=$(func_parser_value "${lines[5]}")
     # parser inference 
-    inference_py=$(func_parser_value "${lines[4]}")
-    use_gpu_key=$(func_parser_key "${lines[5]}")
-    use_gpu_list=$(func_parser_value "${lines[5]}")
-    use_mkldnn_key=$(func_parser_key "${lines[6]}")
-    use_mkldnn_list=$(func_parser_value "${lines[6]}")
-    cpu_threads_key=$(func_parser_key "${lines[7]}")
-    cpu_threads_list=$(func_parser_value "${lines[7]}")
-    batch_size_key=$(func_parser_key "${lines[8]}")
-    batch_size_list=$(func_parser_value "${lines[8]}")
-    use_trt_key=$(func_parser_key "${lines[9]}")
-    use_trt_list=$(func_parser_value "${lines[9]}")
-    precision_key=$(func_parser_key "${lines[10]}")
-    precision_list=$(func_parser_value "${lines[10]}")
-    infer_model_key=$(func_parser_key "${lines[11]}")
-    image_dir_key=$(func_parser_key "${lines[12]}")
-    infer_img_dir=$(func_parser_value "${lines[12]}")
-    save_log_key=$(func_parser_key "${lines[13]}")
-    benchmark_key=$(func_parser_key "${lines[14]}")
-    benchmark_value=$(func_parser_value "${lines[14]}")
-    infer_key1=$(func_parser_key "${lines[15]}")
-    infer_value1=$(func_parser_value "${lines[15]}")
+    inference_py=$(func_parser_value "${lines[6]}")
+    use_gpu_key=$(func_parser_key "${lines[7]}")
+    use_gpu_list=$(func_parser_value "${lines[7]}")
+    use_mkldnn_key=$(func_parser_key "${lines[8]}")
+    use_mkldnn_list=$(func_parser_value "${lines[8]}")
+    cpu_threads_key=$(func_parser_key "${lines[9]}")
+    cpu_threads_list=$(func_parser_value "${lines[9]}")
+    batch_size_key=$(func_parser_key "${lines[10]}")
+    batch_size_list=$(func_parser_value "${lines[10]}")
+    use_trt_key=$(func_parser_key "${lines[11]}")
+    use_trt_list=$(func_parser_value "${lines[11]}")
+    precision_key=$(func_parser_key "${lines[12]}")
+    precision_list=$(func_parser_value "${lines[12]}")
+    infer_model_key=$(func_parser_key "${lines[13]}")
+    image_dir_key=$(func_parser_key "${lines[14]}")
+    infer_img_dir=$(func_parser_value "${lines[14]}")
+    save_log_key=$(func_parser_key "${lines[15]}")
+    benchmark_key=$(func_parser_key "${lines[16]}")
+    benchmark_value=$(func_parser_value "${lines[16]}")
+    infer_key1=$(func_parser_key "${lines[17]}")
+    infer_value1=$(func_parser_value "${lines[17]}")
 fi
 
 LOG_PATH="./test_tipc/output"
-- 
GitLab