python logger

保存代码运行的日志

import logging


def get_logger(filename, verbosity=1, name=None):
    level_dict = {0: logging.DEBUG, 1: logging.INFO, 2: logging.WARNING}
    formatter = logging.Formatter(
        "[%(asctime)s][%(filename)s][line:%(lineno)d][%(levelname)s] %(message)s"
    )
    logger = logging.getLogger(name)
    logger.setLevel(level_dict[verbosity])

    fh = logging.FileHandler(filename, "w")
    fh.setFormatter(formatter)
    logger.addHandler(fh)

    sh = logging.StreamHandler()
    sh.setFormatter(formatter)
    logger.addHandler(sh)

    return logger


logger = get_logger('./exp.txt')

logger.info('start training!')
MAX_EPOCH = 10
for epoch in range(MAX_EPOCH):
    loss = 0.01
    acc = 0.90
    print('hello')
    logger.info('Epoch:[{}/{}]\t loss={:.5f}\t acc={:.3f}'.format(epoch , MAX_EPOCH, loss, acc ))

logger.info('finish training!')

主要参考了知乎文章 https://www.zhihu.com/question/361602016