tensorflow - 如何在 Keras 上向 Tensorboard 添加文本摘要？

Question

如何在使用 Keras 时向 Tensorboard 添加文本摘要？

我已经设置了 Tensorboard 回调，但我不知道如何添加文本摘要。

例如，我想将运行中使用的不同参数的文本摘要添加到 Tensorboard 中，以便在我重新访问运行日志时不会丢失文档和丢失。

一个选项似乎是将所有参数详细信息包含到日志文件目录名称中，但这看起来很累。

我怎样才能更好地解决这个问题？

Answer 1

对于寻找此内容的其他人，我最终编写了一个扩展 Keras Tensorboard 的自定义回调。稍后我可能会记录比摘要更多的内容，我们可以扩展它以记录更多关于不同事件的内容。我从 another question - to add plot 简化了

from keras.callbacks import TensorBoard
import tensorflow as tf

class LoggingTensorBoard(TensorBoard):    

    def __init__(self, log_dir, settings_str_to_log, **kwargs):
        super(LoggingTensorBoard, self).__init__(log_dir, **kwargs)

        self.settings_str = settings_str_to_log

    def on_train_begin(self, logs=None):
        TensorBoard.on_train_begin(self, logs=logs)

        tensor =  tf.convert_to_tensor(self.settings_str)
        summary = tf.summary.text ("Run Settings", tensor)

        with  tf.Session() as sess:
            s = sess.run(summary)
            self.writer.add_summary(s)

创建此 Tensorboard 的实例并添加到 model.fit 回调，您的 settings_str_to_log 将显示在 Tensorboard 的文本选项卡上。