pytorch中网络训练添加日志

pytorch中添加日志

如果是指保存训练过程中的loss，accuracy等metric的话，可以采用下面的方法：

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

可以在输出到终端的同时输出到filename文件中。

使用时在训练脚本train.py中写入：

logger = get_logger('/path/to/exp/exp.log')
 
logger.info('start training!')
for epoch in range(MAX_EPOCH):
    ...
    loss