未知层：当我尝试 load_model 时的 KerasLayer

Question

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

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

然后再次加载模型

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

我收到以下错误

ValueError：未知层：KerasLayer

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

Answer 1

我刚刚找到了适合我的解决方案

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

Unknown layer: KerasLayer when i try to load_model