BERT-pytorch icon indicating copy to clipboard operation
BERT-pytorch copied to clipboard

Update retrain.py

Open watseob opened this issue 4 years ago • 0 comments

Predicting 'is_next' seems incorrect. Since the 'is_next' label value consists of 0 and 1, ignore_index should be 'None'.

watseob avatar May 14 '20 10:05 watseob