微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

如何加载从 TFHub 下载的 SavedModel 作为 Keras 模型?

如何解决如何加载从 TFHub 下载的 SavedModel 作为 Keras 模型?

我对 tensorflow 真的很陌生,我正在尝试弄清楚如何加载从 TFHub 下载的 SavedModel 作为 Keras 模型。

我从 https://tfhub.dev/google/LaBSE/2 下载了 SavedModel。我能够使用 tf.saved_model.load() 在我的代码中加载模型。但是,当我尝试使用 tf.keras.models.load_model 加载模型时,出现以下错误

索引错误:列表索引 (0) 超出范围

想知道如何将模型加载或转换为 Keras 模型?

谢谢!

解决方法

您是否检查过中心中的示例使用部分?那里的示例代码对您有用吗? 请尝试使用以下代码段直接从 Hub 加载模型:

hub.KerasLayer("https://tfhub.dev/google/LaBSE/2")

因为你的是一个集线器模型,上面的行应该适合你。


但是,如果您想直接将 SavedModel (.pb) 转换为 Keras 模型,请选中 here

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 [email protected] 举报,一经查实,本站将立刻删除。