如何加载从 TFHub 下载的 SavedModel 作为 Keras 模型？答案

【问题标题】：How to load a SavedModel downloaded from TFHub as a Keras model?如何加载从 TFHub 下载的 SavedModel 作为 Keras 模型？
【发布时间】：2021-07-30 04:34:33
【问题描述】：

我是 tensorflow 的新手，我正在尝试弄清楚如何加载从 TFHub 下载的 SavedModel 作为 Keras 模型。

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

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

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

谢谢！

【问题讨论】：

标签： tensorflow keras tf.keras

【解决方案1】：

您是否检查过 Hub 中的示例使用部分？那里的示例代码对您有用吗？请尝试以下 sn-p 直接从 Hub 加载模型：

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

因为你的是 Hub 模型，所以上面的行应该适合你。

但是，如果您想直接将 SavedModel (.pb) 转换为 Keras 模型，请查看here。

【讨论】：