当我尝试将模型保存为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
答案 0 :(得分：4)
我刚刚找到了对我有用的解决方案
my_reloaded_model = tf.keras.models.load_model((path),custom_objects={'KerasLayer':hub.KerasLayer})