diff --git a/tools/export_model.py b/tools/export_model.py index cae87aca129134d64711e364bf10428d69500a06..71ecc63b0bdadce8b2bd41dc9119ab556aaa435c 100755 --- a/tools/export_model.py +++ b/tools/export_model.py @@ -26,7 +26,7 @@ from paddle.jit import to_static from ppocr.modeling.architectures import build_model from ppocr.postprocess import build_post_process -from ppocr.utils.save_load import init_model +from ppocr.utils.save_load import load_dygraph_params from ppocr.utils.logging import get_logger from tools.program import load_config, merge_config, ArgsParser @@ -99,7 +99,7 @@ def main(): else: # base rec model config["Architecture"]["Head"]["out_channels"] = char_num model = build_model(config["Architecture"]) - init_model(config, model) + _ = load_dygraph_params(config, model, logger, None) model.eval() save_path = config["Global"]["save_inference_dir"]