cuda error: device-side assert triggered问题解决

在利用BertForSequenceClassification训练下游任务时，遇到了cuda error: device-side assert triggered问题，通过查阅资料，大家普遍认可的做法是将模型切换至cpu进行运行，以输出真正的错误，但是在我用Jupyter notebook测试时内核频繁崩溃，无法输出任何错误。后来发现有人提到分类标签超出了既定的范围，发现没有设置BertForSequenceClassification的分类标签数量。

BertForSequenceClassification的分类标签可以通过BertConfig设置，即config.num_labels=num_classes

参考链接：

1、关于cuda error:device-side assert triggered的3种解决办法，关于cuda error:device-side assert triggered的3种解决办法 - 知乎；

2、Pytorch报错 CUDA error: device-side assert triggered，Pytorch报错 CUDA error: device-side assert triggered_cunn_classnllcriterion_updateoutput_kernel: block:-CSDN博客3、完美解决-RuntimeError: CUDA error: device-side assert triggered，完美解决-RuntimeError: CUDA error: device-side assert triggered-腾讯云开发者社区-腾讯云