RuntimeError: CUDA error: device-side assert triggered
目录项目场景:问题描述总结项目场景:当我使用AST作为预训练模型,并将输出变成十分类时,出现了下面的问题。模型训练时出现了下面的问题(这里给出报错的部分截图): 当时找了几篇文章都说是在做分类任务时,训练数据中存在超出分类数目的标签。但是我检查过,分类数目和标签是对应的。后来找到了RuntimeError: CUDA error: device-side assert triggered的解决_思念殇千寻的博客-CSDN博客 当时就根据作者的总结:
发现我使用的是BCELoss,但是AST
原创
2022-07-09 09:55:16 ·
529 阅读 ·
0 评论