Browse Source

display model info

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

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

@@ -872,6 +872,7 @@ void GetGroupName(ge::proto::ModelDef &model_def) {
FMK_FUNC_HOST_VISIBILITY void PrintModelInfo(ge::proto::ModelDef *model_def) {
std::cout << "------------ Display Model Info start ------------" << std::endl;

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

// resource info
auto model_attr_map = model_def->mutable_attr();
auto 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);


Loading…
Cancel
Save