如何在Keras上的Tensorboard中添加文本摘要？

Question

How do I add text summary to Tensorboard when using Keras?

I have setup a Tensorboard callback, but I am lost on how to add a text summary.

For example, I would like to add a text summary of different parameters used in the run into the Tensorboard so that for documentation and not lost when I revisit the run log.

One option seems to be to include all the parameter details to the logfile dir name, but that looks exhausting.

How can I better solve this problem?

Answer 1

对于寻找此功能的其他人，我最后编写了一个自定义回调，扩展了Keras Tensorboard。稍后我可能会记录比摘要更多的内容，我们可以将其扩展为在不同事件上记录更多内容。我从另一个问题简化了-添加情节

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的“文本”选项卡中。