BERT-pytorch
BERT-pytorch copied to clipboard
Update retrain.py
Predicting 'is_next' seems incorrect. Since the 'is_next' label value consists of 0 and 1, ignore_index should be 'None'.