跑深度学习模型进行的日志输出

跑深度学习模型进行的日志输出

 

构造Logger类进行输出重定向。

在write函数中，使用两个函数使得命令行和日志都有输出

class Logger(object):
    def __init__(self, filename='default.log', stream=sys.stdout):
	    self.terminal = stream
	    self.log = open(filename, 'a')

    def write(self, message):
	    self.terminal.write(message)
	    self.log.write(message)

    def flush(self):
	    pass

sys.stdout = Logger('imitation_network_model.log', sys.stdout)