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

2024-05-20 • 问答

我对 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。

本文链接：https://www.f2er.com/12217.html