pytorch cuda runtime error (59) 错误


pytorch训练过程中:出现

cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/generic/THCTensorCopy.c:70

输出Variable(label).cuda()为空,在变成Variable之前 输出label最大最小值,一般情况绝对是超出了标签类别数。

猜你喜欢

转载自blog.csdn.net/leilei18a/article/details/80670682