python - 如何在训练期间记录给定层的激活以供以后统计？

Question

我正在 tf.keras 中训练一个模型，并且我想在训练期间保存给定层的所有激活（例如在每个批次中），以便能够在 Tensorboard 中跟踪这些激活的箱线图/直方图。

我在无法用于此目的的 Tensorboard 回调选项之间迷路了。

我曾尝试编写自定义回调，但.numpy在model.layers[i].output.

我也尝试过自定义指标，但从示例中可以看出它们只存储一个带有shape=().

我找到了关于可视化推理激活的答案，但在训练数据本身的训练期间却没有。

谢谢

标签： pythontensorflowkerastensor