ZhuiyiTechnology / WoBERT

以词为基本单位的中文BERT
Apache License 2.0
455 stars 69 forks source link

CUDA error: device-side assert triggered #5

Closed JeremySun1224 closed 3 years ago

JeremySun1224 commented 3 years ago

您好,我尝试在vocab.txt词表中添加了“家居”和“时政”两个词表文件中没有的词汇后,使用模型时报了这样的错: CUDA error: device-side assert triggered。看到issue里也有遇到同样问题的朋友,回复是在bert4keras里边可以通过compound_tokens的方式增加新词。由于使用前我把TF版的WoBERT转成PyTorch版的了,请问我应该怎么修改呢,您有什么建议么?由衷感谢🌹