Merge branch 'fix-log-display-kingsleyl' into 'v3.5-integration'

Fix the log display of Model.summary() See merge request dl/hugectr/hugectr!696
LeetcodeCourageK · Mar 1, 2022 · 0bc1c8a · 0bc1c8a
2 parents c8e5d75 + 4452a5f
commit 0bc1c8a
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/HugeCTR/src/pybind/model.cpp b/HugeCTR/src/pybind/model.cpp
@@ -1060,11 +1060,10 @@ void Model::summary() {
   for (auto tensor_entry : train_tensor_entries_list_[0]) {
     tensor_shape_info_.insert(std::make_pair(tensor_entry.name, tensor_entry.bag.get_dimensions()));
   }
-
+  HCTR_PRINT(INFO,
+             "===================================================Model "
+             "Summary===================================================\n");
   auto log = HCTR_LOG_S(INFO, ROOT);
-  log << "===================================================Model "
-         "Summary==================================================="
-      << std::endl;
   log << std::left << std::setw(40) << std::setfill(' ') << "label" << std::left << std::setw(30)
       << std::setfill(' ') << "Dense" << std::left << std::setw(30) << std::setfill(' ') << "Sparse"
       << std::endl;