RuntimeError: CUDA error: device-side assert triggered

 跑了三轮迭代后，死掉了，遇到上面的bug,

 

/opt/conda/conda-bld/pytorch_1614378063927/work/aten/src/THCUNN/ClassNLLCriterion.cu:59: ClassNLLCriterion_updateOutput_no_reduce_kernel: block: [0,0,0], thread: [0,0,0] Assertion `cur_target >= 0 && cur_target < n_classes` failed.

跑的GitHub的模型，用官方给的数据集可以正确出结果，换成我自己的数据集开始报上边的错，

百度到的结果大概分三类:

loss  有的会出 现loss为负数的情况，这样在totch.log中加上

+1e-5torch.sum(weight)+1e-5

但是这种 方法对我不管用

参考链接：

Pytorch问题：device-side assert triggered 未必是标签越界导致_shuozhang_Hebut的博客-优快云博客_/pytorch/aten/src/aten/native/cuda/scattergatherke

btchsize  大小不对应，尝试改变batch_size大小，也不对

标签不匹配（ 我是这个问题）

        参考链接：Pytorch问题：device-side assert triggered 未必是标签越界导致_shuozhang_Hebut的博客-优快云博客_/pytorch/aten/src/aten/native/cuda/scattergatherke