把tensorflow logging输出到文件里面去

今天在重构代码的时候,需要把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

 

你可能感兴趣的:(tensorflow,python学习)