Keras load_mode ValueError: Unknown layer: TokenEmbedding_unknown layer: tokenembedding. please ensure this

作者：Cpp五条 | 2024-04-28 01:48:45

踩

unknown layer: tokenembedding. please ensure this object is passed to the `c

Keras load_mode ValueError: Unknown layer: TokenEmbedding

出现这个问题的原因是因为模型使用了自定义层。具体到我的代码的原因是因为我使用了keras_bert的预训练模型。我们要把我们自定义层的类名和具体的类都作为custom_objects 参数传入load_model方法中。具体到我本人的代码如下：

from keras.models import Model
from keras.models import load_model
from keras_bert import get_custom_objects
custom_objects = get_custom_objects()
model = load_model('bert_150.h5',custom_objects = custom_objects)
1
2
3
4
5

这样就可以了

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/Cpp五条/article/detail/499820