如何将TensorFlow日志重定向到文件? [英] How to redirect TensorFlow logging to a file?

问题描述

我正在使用TensorFlow-Slim，其中有一些有用的日志记录由tf.logging打印到控制台.我想将这些日志记录重定向到文本文件，但是找不到这样做的方法.我查看了tf_logging.py源代码，该源代码公开了以下内容，但似乎没有选择将日志写入文件的选项.如果我错过了什么，请告诉我.

I'm using TensorFlow-Slim, which has some useful logging printed out to console by tf.logging. I would like to redirect those loggings to a text file, but couldn't find a way doing so. I looked at the tf_logging.py source code, which exposes the following, but doesn't seem to have the option to write logs to a file. Please let me know if I missed something.

__all__ = ['log', 'debug', 'error', 'fatal', 'info', 'warn', 'warning',
           'DEBUG', 'ERROR', 'FATAL', 'INFO', 'WARN',
           'flush', 'log_every_n', 'log_first_n', 'vlog',
           'TaskLevelStatusMessage', 'get_verbosity', 'set_verbosity']

推荐答案

import logging

# get TF logger
log = logging.getLogger('tensorflow')
log.setLevel(logging.DEBUG)

# create formatter and add it to the handlers
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')

# create file handler which logs even debug messages
fh = logging.FileHandler('tensorflow.log')
fh.setLevel(logging.DEBUG)
fh.setFormatter(formatter)
log.addHandler(fh)

我的解决方案受到的启发此线程.

这篇关于如何将TensorFlow日志重定向到文件?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！