今天在重构代码的时候,需要把tensorflow logging模块的信息输出到文件中,这里把代码分享以下:
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训练代码里面,运行的时候就会输出一个tesnorflow.log文件,控制台的信息都在里面。
参考文献
[1]. How to redirect TensorFlow logging to a file?. https://stackoverflow.com/questions/40559667/how-to-redirect-tensorflow-logging-to-a-file