BERT_NER.py#L450 same code in if-else in both branches

kyzhouhzau / BERT-NER

Use Google's BERT for named entity recognition （CoNLL-2003 as the dataset）.

MIT License

1.23k stars 335 forks source link

BERT_NER.py#L450 same code in if-else in both branches #91

Open natasasdj opened 3 years ago

natasasdj commented 3 years ago

BERT_NER.py#L450

if FLAGS.crf: (total_loss, logits,predicts) = create_model(bert_config, is_training, input_ids, mask, segment_ids, label_ids,num_labels, use_one_hot_embeddings)`

else: (total_loss, logits, predicts) = create_model(bert_config, is_training, input_ids, mask, segment_ids, label_ids,num_labels, use_one_hot_embeddings)

The same code is for both conditions.