提交 77ead937 编写于 作者: M Megvii Engine Team 提交者: huangxinda

fix(src/serialization): fix compatibility error of oss model

GitOrigin-RevId: 43e0fa4fe14cbfdcbc835a35487368098e9772ed
上级 070c8117
......@@ -309,7 +309,7 @@ def optimize_for_inference(args, outputs):
"enable_chwn4",
"enable_fuse_conv_bias_nonlinearity",
"enable_fuse_conv_bias_with_z",
"eaable_fuse_preprocess",
"enable_fuse_preprocess",
]
kwargs = {}
for k in args_list:
......
......@@ -723,15 +723,17 @@ GraphLoaderOSS::OprLoadContextImpl::load_tensor_shared() {
Metadata GraphLoaderOSS::OprLoadContextImpl::load_metadata() {
const auto* fbmeta = m_loader->m_graph->metadata();
Metadata ret;
ret.is_valid = fbmeta->is_valid();
ret.graph_modified = fbmeta->graph_modified();
if (fbmeta->user_info()) {
ret.user_info = fbmeta->user_info()->str();
ret.has_user_info = true;
}
if (fbmeta->optimize_options()) {
ret.optimize_options = fbmeta->optimize_options();
ret.optimized_for_inference = true;
if (fbmeta) {
ret.is_valid = fbmeta->is_valid();
ret.graph_modified = fbmeta->graph_modified();
if (fbmeta->user_info()) {
ret.user_info = fbmeta->user_info()->str();
ret.has_user_info = true;
}
if (fbmeta->optimize_options()) {
ret.optimize_options = fbmeta->optimize_options();
ret.optimized_for_inference = true;
}
}
return ret;
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册