如何解决如何加载从 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] 举报,一经查实,本站将立刻删除。