Browse Source

fix bug for infer failed at 310

Signed-off-by: zhupuxu <zhupuxu@huawei.com>
pull/16097/head
zhupuxu 4 years ago
parent
commit
31a31004ff
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/cxx_api/graph/graph_data.cc

+ 1
- 1
mindspore/ccsrc/cxx_api/graph/graph_data.cc View File

@@ -41,7 +41,7 @@ Graph::GraphData::GraphData(Buffer om_data, enum ModelType model_type)
ge::ModelData model_data;
model_data.model_data = om_data.MutableData();
model_data.model_len = om_data.DataSize();
ge::Status ret = helper.LoadModel(model_data);
ge::Status ret = helper.LoadRootModel(model_data);
if (ret != ge::SUCCESS) {
MS_LOG(EXCEPTION) << "Invalid input data cannot parse to om.";
}


Loading…
Cancel
Save