def _get_logger(self, name):
old_class = logging.getLoggerClass()
logging.setLoggerClass(MPILogger.CustomLogger)
logger = logging.getLogger(name)
logging.setLoggerClass(old_class)
return logger
###########################################################################
# Standard logging functions. Log only at mpi rank 0.
评论列表
文章目录