在批量使用Keras系列时，向张力板添加直方图摘要

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

---

但我需要在Tensorboard中存储直方图、激活和权重的图像。有什么方法可以做到这一点吗？

您可以通过调用

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)

---

请参阅此处@UpasanaMittal的精确副本，该链接仅用于标量摘要。我正在寻找直方图和图像存储在Tensorboard的摘要中。我需要在批处理中使用model.train\u时使用Keras