在使用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)