gpt4 book ai didi

tensorflow - 如何为 Tensorflow Hub 模块的特定输入获取所有层的激活?

转载 作者:行者123 更新时间:2023-12-03 12:38:17 26 4
gpt4 key购买 nike

我是 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 版本)碰巧只导出最后一层,因此没有适当支持的方式来获取其他层。 (也就是说,您始终可以通过检查图形并按名称选择张量来进行试验,但这确实存在停止使用较新的模块或库版本的风险。)

关于tensorflow - 如何为 Tensorflow Hub 模块的特定输入获取所有层的激活?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/52304359/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com