python - 未知层 : KerasLayer when i try to load_model

Question

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

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

然后再次加载模型

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

我收到以下错误
值错误:未知层:KerasLayer
有什么帮助吗？
我正在使用
tensorflow 版本:2.2.0
keras 版本:2.3.0-tf

Answer 1

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

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