Ass*_*led 3 python keras tensorflow2.0
当我尝试将我的模型保存为 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
Ass*_*led 10
我刚刚找到了一个对我有用的解决方案
my_reloaded_model = tf.keras.models.load_model((path),custom_objects={'KerasLayer':hub.KerasLayer})
| 归档时间： |
|
| 查看次数： |
1244 次 |
| 最近记录： |