RuntimeError: CUDA error: device-side assert triggered

lonePatient / BERT-NER-Pytorch

Chinese NER(Named Entity Recognition) using BERT(Softmax, CRF, Span)

MIT License

2.08k stars 427 forks source link

Open allenyummy opened 4 years ago

allenyummy commented 4 years ago

您好，

用單顆 gpu 進行 brert-crf-ner 的訓練，出現如標題的錯誤，想請問該如何解決？

謝謝！

allenyummy commented 4 years ago

後來發現是因為一些special token: [CLS], [SEP], [PAD] 經過一番處理後是 -100，為的是不讓這些貢獻到 loss function 上，想請問這些 special token 應該改成什麼呢？

直接改成 O 嗎？