Open skywindy opened 4 years ago
提问时请尽可能提供如下信息: 苏老师,您好,我用bert4keras加载albert模型报以下错误: tensorflow.python.framework.errors_impl.NotFoundError: Key bert/encoder/embedding_hidden_mapping_in/kernel not found in checkpoint 使用了brightmart训练的albert_base版,和您这边转换后的版本,都有这个错误
传入model='albert'了吗?认真看过examples了吗?
model='albert'
model = build_transformer_model( config_path, checkpoint_path, model='albert' )
brightmart版不支持,如果是我自己转换的版本,则没有embedding_hidden_mapping_in层,因此也不会保证这个错。
提问时请尽可能提供如下信息: 苏老师,您好,我用bert4keras加载albert模型报以下错误: tensorflow.python.framework.errors_impl.NotFoundError: Key bert/encoder/embedding_hidden_mapping_in/kernel not found in checkpoint 使用了brightmart训练的albert_base版,和您这边转换后的版本,都有这个错误
基本信息