diff --git a/deploy/slim/quant_post_static.py b/deploy/slim/quant_post_static.py index edab53d398fa3a2515c4881ea54a458edb8a553c..5c8469794ad29e18dad15f985b611e423fd4b474 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 e55d612313f25ea06d428711facdc4716adfbdec..ec20912a5f1fd73a33837eaa457b04202c38e08a 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