未验证 提交 45a3c656 编写于 作者: W WangZhen 提交者: GitHub

[JitLayer]Save property meta file to correct path (#46131) (#46195)

上级 d1ce974e
...@@ -1090,8 +1090,9 @@ def save(layer, path, input_spec=None, **configs): ...@@ -1090,8 +1090,9 @@ def save(layer, path, input_spec=None, **configs):
ordered_vars)), ordered_vars)),
filename=params_filename) filename=params_filename)
# save property # save property
property_filename = file_prefix + INFER_PROPERTY_SUFFIX property_save_path = os.path.join(os.path.normpath(model_path),
_save_property(property_filename, property_vals) file_prefix + INFER_PROPERTY_SUFFIX)
_save_property(property_save_path, property_vals)
# NOTE(chenweihang): [ Save extra variable info ] # NOTE(chenweihang): [ Save extra variable info ]
# save_inference_model will lose some important variable information, including: # save_inference_model will lose some important variable information, including:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册