如何加载从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中的示例使用部分?这里的示例代码对你有用吗?请尝试下面的代码片段直接从Hub加载模型:

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

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

---

但是，如果您想直接将SavedModel (.pb)转换为Keras模型，则请点击此处。