From 0cc9870eb32a584bed79ca243c41ed960414d680 Mon Sep 17 00:00:00 2001 From: zhangyubo0722 <94225063+zhangyubo0722@users.noreply.github.com> Date: Mon, 12 May 2025 21:22:52 +0800 Subject: [PATCH] fix pdmodel to json (#15122) Co-authored-by: zhangyubo0722 --- ppocr/utils/save_load.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/ppocr/utils/save_load.py b/ppocr/utils/save_load.py index cc8bc468ce..df0223d706 100644 --- a/ppocr/utils/save_load.py +++ b/ppocr/utils/save_load.py @@ -20,6 +20,7 @@ import errno import os import pickle import json +from packaging import version import paddle @@ -288,7 +289,8 @@ def update_train_results(config, prefix, metric_info, done_flag=False, last_num= config["Global"]["save_model_dir"], "train_result.json" ) save_model_tag = ["pdparams", "pdopt", "pdstates"] - if FLAGS_json_format_model: + paddle_version = version.parse(paddle.__version__) + if FLAGS_json_format_model or paddle_version >= version.parse("3.0.0"): save_inference_files = { "inference_config": "inference.yml", "pdmodel": "inference.json",