如何在事件文件中一次只保留一个摘要?

时间:2017-07-03 20:09:08

标签: python tensorflow tensorboard

我有一种情况,我一次只想存储一个张量。在IIUC中,FileWriter将摘要附加到现有事件文件中。现在,每次我想写一个摘要时,我都会在init类中执行以下操作:

self.WRITER = tf.summary.FileWriter(self.LOGDIR, 
                                    max_queue=1, 
                                    flush_secs=9999999)

并在类方法中:

summary = self.SESSION.run(tf.summary.tensor_summary('frame', 
                                                     image_tensor))
self.WRITER.add_summary(summary)
self.WRITER.flush()
self.WRITER.close()

使用.close(),它只会写一次。如果没有.close(),它会附加到事件文件中。在方法开头使用self.WRITER.reopen()时,它会添加新的事件文件。我希望每次都覆盖一个事件文件。

有没有办法通过TensorFlow执行此操作,还是需要手动删除旧文件并在每次迭代时创建新摘要?

1 个答案:

答案 0 :(得分:0)

我看到没有办法用最后一个事件刷新文件,但你可以使用

os.remove("path/to/file")

每次迭代。