tensorflow 保存日志

转载请注明出处：http://www.javashuo.com/article/p-gqgvpktz-ky.html

import logging
import time

# 建立一个logger
logger = logging.getLogger('mylogger')
logger.setLevel(logging.DEBUG)

# 建立一个handler，用于写入日志文件
timestamp = str(int(time.time()))
fh = logging.FileHandler('./log/log_' + timestamp +'.txt')
fh.setLevel(logging.DEBUG)

# 再建立一个handler，用于输出到控制台
ch = logging.StreamHandler()
ch.setLevel(logging.DEBUG)

# 定义handler的输出格式
formatter = logging.Formatter('[%(asctime)s][%(levelname)s] ## %(message)s')
fh.setFormatter(formatter)
ch.setFormatter(formatter)

# 给logger添加handler
logger.addHandler(fh)
logger.addHandler(ch)

# 将print函数改成logger.info函数就能够将输出信息保存到日志文件中
logger.info('load data')