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

Question

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

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

Answer 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)