keras_bert加载训练好的模型报错 ValueError: Unknown layer: TokenEmbedding

使用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())