重定向 --- 同时在中断和文件进行输出

重定向 --- 同时在中断和文件进行输出

 

使用重定向技术，在模型训练的过程中同时在中断和日志文件中输出我们的结果。

 

代码：

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('filename.log', sys.stdout)