Browse Source

display model info

pull/688/head
wangwenhua1@huawei.com 5 years ago
parent
commit
500c35d3a9
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      ge/session/omg.cc

+ 2
- 2
ge/session/omg.cc View File

@@ -874,7 +874,7 @@ FMK_FUNC_HOST_VISIBILITY void PrintModelInfo(ge::proto::ModelDef *model_def) {

auto model_attr_map = model_def->mutable_attr();
// system info
iter = model_attr_map->find(ATTR_MODEL_ATC_VERSION);
auto iter = model_attr_map->find(ATTR_MODEL_ATC_VERSION);
auto atc_version = (iter != model_attr_map->end()) ? iter->second.s() : "";
iter = model_attr_map->find("soc_version");
auto soc_version = (iter != model_attr_map->end()) ? iter->second.s() : "";
@@ -889,7 +889,7 @@ FMK_FUNC_HOST_VISIBILITY void PrintModelInfo(ge::proto::ModelDef *model_def) {
<< "[" << framework_type << "]." << std::endl;

// resource info
auto iter = model_attr_map->find(ATTR_MODEL_MEMORY_SIZE);
iter = model_attr_map->find(ATTR_MODEL_MEMORY_SIZE);
auto memory_size = (iter != model_attr_map->end()) ? iter->second.i() : -1;
iter = model_attr_map->find(ATTR_MODEL_WEIGHT_SIZE);
auto weight_size = (iter != model_attr_map->end()) ? iter->second.i() : -1;


Loading…
Cancel
Save