display model info

ge/session/omg.cc

@@ -900,9 +900,9 @@ FMK_FUNC_HOST_VISIBILITY void PrintModelInfo(ge::proto::ModelDef *model_def) {
   auto event_num = (iter != model_attr_map->end()) ? iter->second.i() : -1;
   std::cout << "resource info: "
             << ATTR_MODEL_MEMORY_SIZE
-            << "[" << memory_size << "], "
+            << "[" << memory_size << " B], "
             << ATTR_MODEL_WEIGHT_SIZE
-            << "[" << weight_size << "], "
+            << "[" << weight_size << " B], "
             << ATTR_MODEL_STREAM_NUM
             << "[" << stream_num << "], "
             << ATTR_MODEL_EVENT_NUM
@@ -914,27 +914,27 @@ FMK_FUNC_HOST_VISIBILITY void PrintModelInfo(ge::proto::ModelDef *model_def) {
   if (iter == model_attr_map->end()) {
     std::cout << "Display Model Info failed, attr \"om_info_list\" is not found in om, check the version is matched."
               << std::endl;
-    std::cout << "------------ Display Model Info end   ------------"  << std::endl;
+    std::cout << "============ Display Model Info end   ============"  << std::endl;
     return;
   }
   auto list_size = iter->second.list().i_size();
   if (list_size == kOmInfoSize) {
     std::cout << "om info: "
               << "modeldef_size"
-              << "[" << iter->second.list().i(0) << "], "
+              << "[" << iter->second.list().i(0) << " B], "
               << "weight_data_size"
-              << "[" << iter->second.list().i(1) << "], "
+              << "[" << iter->second.list().i(1) << " B], "
               << "tbe_kernels_size"
-              << "[" << iter->second.list().i(2) << "], "
+              << "[" << iter->second.list().i(2) << " B], "
               << "cust_aicpu_kernel_store_size"
-              << "[" << iter->second.list().i(3) << "], "
+              << "[" << iter->second.list().i(3) << " B], "
               << "task_info_size"
-              << "[" << iter->second.list().i(4) << "]." << std::endl;
+              << "[" << iter->second.list().i(4) << " B]." << std::endl;
   } else {
     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;
 }
 
 FMK_FUNC_HOST_VISIBILITY Status ConvertOm(const char *model_file, const char *json_file, bool is_covert_to_json) {