提交 b9440ce5 编写于 作者: D dongshuilong

update wholechain klquant

上级 62b3775e
...@@ -41,11 +41,11 @@ def main(): ...@@ -41,11 +41,11 @@ def main():
'inference.pdmodel')) and os.path.exists( 'inference.pdmodel')) and os.path.exists(
os.path.join(config["Global"]["save_inference_dir"], os.path.join(config["Global"]["save_inference_dir"],
'inference.pdiparams')) 'inference.pdiparams'))
config["DataLoader"]["Train"]["sampler"]["batch_size"] = 1 config["DataLoader"]["Eval"]["sampler"]["batch_size"] = 1
config["DataLoader"]["Train"]["loader"]["num_workers"] = 0 config["DataLoader"]["Eval"]["loader"]["num_workers"] = 0
init_logger() init_logger()
device = paddle.set_device("cpu") device = paddle.set_device("cpu")
train_dataloader = build_dataloader(config["DataLoader"], "Train", device, train_dataloader = build_dataloader(config["DataLoader"], "Eval", device,
False) False)
def sample_generator(loader): def sample_generator(loader):
......
...@@ -156,7 +156,7 @@ function func_inference(){ ...@@ -156,7 +156,7 @@ function func_inference(){
done done
} }
if [ ${MODE} = "whole_infer" ] || [${MODE} = "klquant_whole_infer" ]; then if [ ${MODE} = "whole_infer" ] || [ ${MODE} = "klquant_whole_infer" ]; then
IFS="|" IFS="|"
infer_export_flag=(${infer_export_flag}) infer_export_flag=(${infer_export_flag})
if [ ${infer_export_flag} != "null" ] && [ ${infer_export_flag} != "False" ]; then if [ ${infer_export_flag} != "null" ] && [ ${infer_export_flag} != "False" ]; then
...@@ -199,7 +199,7 @@ elif [ ${MODE} = "klquant_whole_infer" ]; then ...@@ -199,7 +199,7 @@ elif [ ${MODE} = "klquant_whole_infer" ]; then
ln -s __params__ inference.pdiparams ln -s __params__ inference.pdiparams
cd ../../deploy cd ../../deploy
is_quant=True 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 .. cd ..
fi fi
else else
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册