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

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

上级 7098bfdf
......@@ -1090,8 +1090,9 @@ def save(layer, path, input_spec=None, **configs):
ordered_vars)),
filename=params_filename)
# save property
property_filename = file_prefix + INFER_PROPERTY_SUFFIX
_save_property(property_filename, property_vals)
property_save_path = os.path.join(os.path.normpath(model_path),
file_prefix + INFER_PROPERTY_SUFFIX)
_save_property(property_save_path, property_vals)
# NOTE(chenweihang): [ Save extra variable info ]
# 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.
先完成此消息的编辑!
想要评论请 注册