From 4452a5f46893fe1a9950d9cacc546c27249351bd Mon Sep 17 00:00:00 2001 From: Kingsley Liu Date: Tue, 1 Mar 2022 01:22:51 -0800 Subject: [PATCH] Fix the log display of Model.summary() --- HugeCTR/src/pybind/model.cpp | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/HugeCTR/src/pybind/model.cpp b/HugeCTR/src/pybind/model.cpp index e3aa8e4245..d76baa70e8 100644 --- 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;