python - 未知层：当我尝试加载模型时 KerasLayer

Question

当我尝试将我的模型保存为 hdf5

path = 'path.h5'
model.save(path)

然后再次加载模型

my_reloaded_model = tf.keras.models.load_model(path)

我收到以下错误

ValueError：未知层：KerasLayer

有什么帮助吗？我正在使用
tensorflow 版本：2.2.0
keras 版本：2.3.0-tf

标签： pythonkerastensorflow2.0

Answer 1

我刚刚找到了一个对我有用的解决方案

my_reloaded_model = tf.keras.models.load_model((path),custom_objects={'KerasLayer':hub.KerasLayer})