diff --git a/include/lbann/callbacks/mlperf_logging.hpp b/include/lbann/callbacks/mlperf_logging.hpp index 2e19046033d..fd1097fc56c 100644 --- a/include/lbann/callbacks/mlperf_logging.hpp +++ b/include/lbann/callbacks/mlperf_logging.hpp @@ -131,11 +131,10 @@ class mlperf_logging : public callback_base { std::string m_sub_division; std::string m_sub_status; std::string m_sub_platform; - //FIXME: Include option to create separate file with just logging data? /* @brief name of output file. Default = results.txt */ //std::string m_output_filename; /* @brief DiHydrogen logger */ - h2::Logger m_logger{":::MLLOG"}; + h2::Logger m_logger{"mlperf_logger", "stdout", ":::MLLOG"}; }; // class mlperf_logging diff --git a/src/callbacks/mlperf_logging.cpp b/src/callbacks/mlperf_logging.cpp index 883404630c0..d842c0f711e 100644 --- a/src/callbacks/mlperf_logging.cpp +++ b/src/callbacks/mlperf_logging.cpp @@ -44,6 +44,7 @@ namespace lbann { namespace callback { namespace { + void print_value(std::ostringstream& os, int value) { os << value; @@ -116,7 +117,7 @@ void mlperf_logging::print(std::ostringstream& os, else os << ", " << "\"epoch_num\": " << epoch << "}}"; - H2_INFO(os.str()); + m_logger.get()->info(os.str()); os.flush(); }