Browse Source

display model info

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

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

@@ -871,7 +871,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;
std::cout << "============ Display Model Info start ============" << std::endl;

auto model_attr_map = model_def->mutable_attr();
// system info
@@ -881,7 +881,7 @@ FMK_FUNC_HOST_VISIBILITY void PrintModelInfo(ge::proto::ModelDef *model_def) {
auto soc_version = (iter != model_attr_map->end()) ? iter->second.s() : "";
iter = model_attr_map->find("framework_type");
auto framework_type = (iter != model_attr_map->end()) ? iter->second.s() : "";
std::cout << "system info: "
std::cout << "system info: "
<< ATTR_MODEL_ATC_VERSION
<< "[" << atc_version << "], "
<< "soc_version"
@@ -919,7 +919,7 @@ FMK_FUNC_HOST_VISIBILITY void PrintModelInfo(ge::proto::ModelDef *model_def) {
}
auto list_size = iter->second.list().i_size();
if (list_size == kOmInfoSize) {
std::cout << "om info: "
std::cout << "om info: "
<< "modeldef_size"
<< "[" << iter->second.list().i(0) << " B], "
<< "weight_data_size"


Loading…
Cancel
Save