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

Question

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

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

标签： kerastensorboard

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)