pytorch训练过程中:出现
cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/generic/THCTensorCopy.c:70
输出Variable(label).cuda()为空,在变成Variable之前 输出label最大最小值,一般情况绝对是超出了标签类别数。
一般情况是 标签出现问题。可能本地存储图片或者矩阵没有问题,但是经过数据增强后 会出现问题。建议输出看看
pytorch训练过程中:出现
cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/generic/THCTensorCopy.c:70
输出Variable(label).cuda()为空,在变成Variable之前 输出label最大最小值,一般情况绝对是超出了标签类别数。
一般情况是 标签出现问题。可能本地存储图片或者矩阵没有问题,但是经过数据增强后 会出现问题。建议输出看看