Update retrain.py - Githubissues

codertimo / BERT-pytorch

Google AI 2018 BERT pytorch implementation

Apache License 2.0

6.09k stars 1.29k forks source link

Open watseob opened 4 years ago

watseob commented 4 years ago

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