BERT-NER-Pytorch icon indicating copy to clipboard operation
BERT-NER-Pytorch copied to clipboard

Published 20 hours ago • verified publisher icon lonePatient

RuntimeError: CUDA error: device-side assert triggered

Open allenyummy opened this issue 4 years ago • 1 comments

您好，

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

謝謝！

allenyummy avatar Aug 12 '20 01:08 allenyummy

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

直接改成 O 嗎？

allenyummy avatar Aug 12 '20 06:08 allenyummy