mirror of
https://github.com/PaddlePaddle/PaddleOCR.git
synced 2026-08-30 17:23:03 +08:00
fix pdmodel to json (#15122)
Co-authored-by: zhangyubo0722 <zangyubo0722@163.com>
This commit is contained in:
@@ -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",
|
||||
|
||||
Reference in New Issue
Block a user