keras_bert加载训练好的模型报错 ValueError: Unknown layer: TokenEmbedding_valueerror: keyword grid_b is not recognized; vali

作者：花生_TL007 | 2024-04-28 01:46:56

踩

valueerror: keyword grid_b is not recognized; valid keywords are ['size', 'w

使用keras_bert训练好模型后加载保存的模型，使用load_model()方法，报错 ValueError: Unknown layer: TokenEmbedding；应该是没有加载之前预训练的bert部分参数，需要在load_model()中添加custom_objects=get_custom_objects() ，添加这部分模型

如：model = load_model(model_path, custom_objects=get_custom_objects())

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