使用 Keras train_on_batch 时将直方图摘要添加到张量板答案

【问题标题】：add histogram summaries to tensorboard while using Keras train_on_batch使用 Keras train_on_batch 时将直方图摘要添加到张量板
【发布时间】：2019-01-10 19:13:40
【问题描述】：

我在 keras 中使用 model.train_on_batch 以便能够以不同方式处理不同批次的输入数据。本质上我不能使用 model.fit

但我需要在 Tensorboard 中存储直方图以及激活和权重的图像。有没有办法做到这一点？

【问题讨论】：

参考这里，gist.github.com/joelthchao/ef6caa586b647c3c032a4f84d52e3a11 与stackoverflow.com/questions/44861149/… 完全相同
@UpasanaMittal 该链接仅用于标量摘要。我正在寻找要存储在 Tensorboard 摘要中的直方图和图像。我需要在使用 model.train_on_batch 时使用 Keras 来完成

标签： keras tensorboard

【解决方案1】：

您可以通过调用summary.histogram 并传递每一层的权重来手动完成，如下所示

with summary_writer.as_default():
  for layer in self.model.layers:
    for weight in layer.weights:
      tf.summary.histogram('weights/{}'.format(layer.name), weight, step=your_step)

【讨论】：