我是Tensorflow Hub的新手。我想使用I3D模块并将该网络微调到另一个数据集,我需要获取最后一个隐藏层以及其他一些层的输出。我想知道是否有办法获得其他层的激活。为I3D提供的唯一签名只是“默认”。我认为应该有一种使用Tensorflow Hub模块轻松获取所有层的输出的方法。

import tensorflow_hub as hub
module = hub.Module("https://tfhub.dev/deepmind/i3d-kinetics-600/1", trainable=False)
logits = module(inp)

这将给我最终的图层输出。如何获得其他层的输出,例如第二个卷积层的输出?

最佳答案

https://tfhub.dev/deepmind/i3d-kinetics-400/1(以及* -600版本)恰好仅导出最后一层,因此没有适当支持的方法来获取其他层。 (也就是说,您始终可以通过检查图形并按名称选择张量来进行实验,但这确实有停止使用较新的模块或库版本的风险。)

10-08 16:26