logging - 如何将 TensorFlow 日志重定向到文件？

Question

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

__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']

Answer 1

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)

我的解决方案的灵感来自 this thread .