tensorflow训练模型时打印日志小技巧

定义方法：

import logging
def get_logger(log_file):
    logger=logging.getLogger(log_file)
    logger.setLevel(logging.DEBUG)
    fh=logging.FileHandler(log_file)
    ch=logging.StreamHandler()
    ch.setLevel(logging.INFO)
    formatter=logging.Formatter("%(asctime)s - %(name)s - %(levelname)s -%(message)s")
    fh.setFormatter(formatter)
    ch.setFormatter(formatter)
    logger.addHandler(fh)
    logger.addHandler(ch)
    return logger
def print_config(config,logger):
    """
    print configuration of the model
    :param config:
    :param logger:
    :return:
    """
    for k,v in config.items():
        logger.info("{}:\t{}.".format(k.ljust(15),v))

方法调用：

log_path=os.path.join('log',FLAGS.log_file)
logger=get_logger(log_path)
print_config(config,logger)