search

ProHiryu / albert-chinese-ner

使用预训练语言模型ALBERT做中文NER
MIT License
467 stars 124 forks source link

AttributeError: 'BertConfig' object has no attribute 'embedding_size' #16

Closed parkourcx closed 4 years ago

parkourcx commented 4 years ago

tensorflow-gpu 1.15，ner任务使用了自行训练的albert，config.json内容如下 "attention_probs_dropout_prob": 0.1, "directionality": "bidi", "hidden_act": "gelu", "hidden_dropout_prob": 0.1, "hidden_size": 768, "initializer_range": 0.02, "intermediate_size": 3072, "max_position_embeddings": 512, "num_attention_heads": 12, "num_hidden_layers": 12, "pooler_fc_size": 768, "pooler_num_attention_heads": 12, "pooler_num_fc_layers": 3, "pooler_size_per_head": 128, "pooler_type": "first_token_transform", "type_vocab_size": 2, "vocab_size": 20974 运行程序是提示AttributeError: 'BertConfig' object has no attribute 'embedding_size'，请问是什么原因？

parkourcx commented 4 years ago

报错在modeling.Python文件中的第178行， embedding_size=config.embedding_size 这句