From b9440ce5eb89a85a3c04cca0057b702adcf672e4 Mon Sep 17 00:00:00 2001
From: dongshuilong <dongshuilong@baidu.com>
Date: Wed, 17 Nov 2021 12:43:38 +0000
Subject: [PATCH] update wholechain klquant

---
 deploy/slim/quant_post_static.py         | 6 +++---
 test_tipc/test_train_inference_python.sh | 4 ++--
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/deploy/slim/quant_post_static.py b/deploy/slim/quant_post_static.py
index edab53d3..5c846979 100644
--- a/deploy/slim/quant_post_static.py
+++ b/deploy/slim/quant_post_static.py
@@ -41,11 +41,11 @@ def main():
                      'inference.pdmodel')) and os.path.exists(
                          os.path.join(config["Global"]["save_inference_dir"],
                                       'inference.pdiparams'))
-    config["DataLoader"]["Train"]["sampler"]["batch_size"] = 1
-    config["DataLoader"]["Train"]["loader"]["num_workers"] = 0
+    config["DataLoader"]["Eval"]["sampler"]["batch_size"] = 1
+    config["DataLoader"]["Eval"]["loader"]["num_workers"] = 0
     init_logger()
     device = paddle.set_device("cpu")
-    train_dataloader = build_dataloader(config["DataLoader"], "Train", device,
+    train_dataloader = build_dataloader(config["DataLoader"], "Eval", device,
                                         False)
 
     def sample_generator(loader):
diff --git a/test_tipc/test_train_inference_python.sh b/test_tipc/test_train_inference_python.sh
index e55d6123..ec20912a 100644
--- a/test_tipc/test_train_inference_python.sh
+++ b/test_tipc/test_train_inference_python.sh
@@ -156,7 +156,7 @@ function func_inference(){
     done
 }
 
-if [ ${MODE} = "whole_infer" ] || [${MODE} = "klquant_whole_infer" ]; then
+if [ ${MODE} = "whole_infer" ] || [ ${MODE} = "klquant_whole_infer" ]; then
    IFS="|"
    infer_export_flag=(${infer_export_flag})
    if [ ${infer_export_flag} != "null" ]  && [ ${infer_export_flag} != "False" ]; then
@@ -199,7 +199,7 @@ elif [ ${MODE} = "klquant_whole_infer" ]; then
 	ln -s __params__ inference.pdiparams
 	cd ../../deploy
 	is_quant=True
-        func_inference "${python}" "${inference_py}" "${infer_model}/quant_post_static_model" "../${LOG_PATH}" "${infer_img_dir}" ${is_quant}
+        func_inference "${python}" "${inference_py}" "${infer_model_dir_list}/quant_post_static_model" "../${LOG_PATH}" "${infer_img_dir}" ${is_quant}
 	cd ..
     fi
 else
-- 
GitLab