Function get_logger

utils/engine/logger.py:105–123

(log_dir=None, log_file=None, rank=0)

def get_logger(log_dir=None, log_file=None, rank=0):
    logger = logging.getLogger()
    logger.setLevel(level=logging.INFO)
    del logger.handlers[:]
    formatter = logging.Formatter("%(asctime)s - %(filename)s[line:%(lineno)d] - %(levelname)s: %(message)s")

    if rank == 0:
        if log_dir and log_file:
            file_handler = logging.FileHandler(log_file)
            file_handler.setLevel(level=logging.INFO)
            file_handler.setFormatter(formatter)
            logger.addHandler(file_handler)

        stream_handler = logging.StreamHandler()
        stream_handler.setLevel(logging.DEBUG)
        stream_handler.setFormatter(formatter)

        logger.addHandler(stream_handler)
    return logger