即使删除后,TensorFlow检查点仍然保留

时间:2019-06-21 01:50:39

标签: python tensorflow tensorboard

我正在使用TensorBoard投影仪来可视化嵌入。目前,我正在试验中,并希望删除创建的可视化文件。但是不幸的是,即使我删除带有检查点的文件夹,TensorBoard仍然可以找到可视化的数据。而且真正奇怪的是,即使我使用--logdir=fake将TensorBoard指向错误甚至不存在的目录,它仍然可以直观地看到这些点。我注意到它指向一个已删除的检查点文件。

经过一些研究,我认为TensorFlow正在恢复检查点。如果是这样,我应该如何将其永久删除。 我的代码如下:

def on_epoch_end(self, epoch, logs={}):
    if epoch % self.viz_every == 0:
        writer = tf.summary.FileWriter(self.logdir)

        preds = self.encoder.predict(self.viz_imgs)
        preds = np.array([pred.flatten() for pred in preds])
        encodings = tf.Variable(preds, name='encodings')

        with tf.Session() as sess:
            saver = tf.train.Saver()
            sess.run(tf.global_variables_initializer())
            saver.save(sess, join(self.logdir, 'em.ckpt'))

            config = projector.ProjectorConfig()
            embedding = config.embeddings.add()
            embedding.tensor_name = encodings.name
            embedding.metadata_path = basename(self.metadata_path)
            projector.visualize_embeddings(writer, config)
        writer.close()

0 个答案:

没有答案