display model info

ge/generator/ge_generator.cc

@@ -556,7 +556,7 @@ void GeGenerator::Impl::DisplayModelInfo(AttrHolder &obj) {
     return;
   }
 
-  std::cout << "------ Display Model Info start ------" << std::endl;
+  std::cout << "------------ Display Model Info start ------------" << std::endl;
   // system info
   std::string atc_version;
   (void)ge::AttrUtils::GetStr(obj, ATTR_MODEL_ATC_VERSION, atc_version);
@@ -610,7 +610,7 @@ void GeGenerator::Impl::DisplayModelInfo(AttrHolder &obj) {
     std::cout << "Display Model Info error, please check!"  << std::endl;
   };
 
-  std::cout << "------ Display Model Info end ------"  << std::endl;
+  std::cout << "------------ Display Model Info end   ------------"  << std::endl;
 }
 
 Status GeGenerator::GenerateModel(const Graph &graph, const string &file_name_prefix, const vector<GeTensor> &inputs,

ge/session/omg.cc

@@ -909,7 +909,7 @@ FMK_FUNC_HOST_VISIBILITY void PrintModelInfo(ge::proto::ModelDef *model_def) {
             << std::endl;
 
   // om info
-  iter = model_attr_map->find("om_info_size");
+  iter = model_attr_map->find("om_info_list");
   auto bt = iter->second.list().i_size();
   if (bt == 5) {
     std::cout << "om info: "