Browse Source

display model info

pull/688/head
wangwenhua1@huawei.com 5 years ago
parent
commit
6031e5576c
1 changed files with 6 additions and 0 deletions
  1. +6
    -0
      ge/session/omg.cc

+ 6
- 0
ge/session/omg.cc View File

@@ -910,6 +910,12 @@ FMK_FUNC_HOST_VISIBILITY void PrintModelInfo(ge::proto::ModelDef *model_def) {

// om info
iter = model_attr_map->find("om_info_list");
if (iter == model_attr_map->end()) {
std::cout << "Display Model Info error, om_info_list is not find in om, please check atc version is matched."
<< std::endl;
std::cout << "------------ Display Model Info end ------------" << std::endl;
return;
}
auto bt = iter->second.list().i_size();
if (bt == 5) {
std::cout << "om info: "


Loading…
Cancel
Save