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

时间:2019-12-05 17:49:27

标签: python tensorflow keras tensor

我正在tf.keras中训练一个模型,我想在训练过程中保存给定层的所有激活(例如,在每个批次中),以便能够在Tensorboard中跟踪这些激活的箱线图/直方图。

我在不想为此目的使用的Tensorboard回调选项之间迷路了。

我尝试编写自定义回调,但是在.numpy上使用model.layers[i].output时出现错误。

我也尝试过自定义指标,但是从示例中看来,它们仅使用shape=()存储变量。

我找到了关于可视化激活推理的答案,而不是关于训练数据本身的训练过程。

谢谢

0 个答案:

没有答案