当我尝试将模型另存为 hdf5 时
path = 'path.h5'
model.save(path)
然后再次加载模型
my_reloaded_model = tf.keras.models.load_model(path)
我收到以下错误
ValueError: Unknown layer: KerasLayer
有什么帮助吗?我在用着
tensorflow version: 2.2.0
keras version: 2.3.0-tf
我刚刚找到了一个适合我的解决方案
my_reloaded_model = tf.keras.models.load_model(
(path),
custom_objects={'KerasLayer':hub.KerasLayer}
)
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)