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

Question

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

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

IndexError: list index (0) out of range

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

谢谢！

Answer 1

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

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

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

---

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