search

kyzhouhzau / BERT-NER

Use Google's BERT for named entity recognition （CoNLL-2003 as the dataset）.
MIT License
1.24k stars 335 forks source link

BertModel is_training should be set to False? #13

Open john-theo opened 5 years ago

john-theo commented 5 years ago

Thanks a lot for the amazing work! However, in BERT_NER.py line 348, you set BertModel's is_training parameter to change alone with the fine-tuning procedure, but as the model is not integrated in the real training model (use the output of BertModel's final layer as the real input). So as far as I'm concerned, this parameter should be set to False to gain more reasonable result.