PyTorch报错：RuntimeError: CUDA error: device-side assert triggered at /pytorch/aten/src/THC/generic

小鱼儿 2022-11-15 14:10 280阅读 0赞

训练模型报错

  1. RuntimeError: cuda runtime error (710) : device-side assert triggered at /pytorch/aten/src/THC/generic/THCTensorMath.cu:29

问题解决

原因：标签（label）越界，一般从0开始

方法：更改你的标签值，从0开始

  1. target_names = ['__label__1', '__label__2', '__label__3', '__label__4', '__label__5']
  2.     for i in range(len(questlist)):
  3.         if i < len(questlist):
  4.             for j in range(len(attrlist)):
  5.                 if j < len(attrlist):
  6.                     if attr_names[j] in attrlist[i]:
  7.                         label = target_names[j]
  8.                         line = " ".join(jieba.cut(questlist[i])) + "\t" + str(int(label[-1])-1) ## 标签值减1
  9.                         linelist.append(line)
  10.                         break
  12. line = " ".join(jieba.cut(questlist[i])) + "\t" + str(int(label[-1])-1) ## 标签值减1

发表评论

表情：
评论列表 （有 0 条评论，280人围观）

还没有评论，来说两句吧...

相关阅读