result = result.scatter_(1, input.cpu(), 1) 报错

问题

“C:\w\1\s\tmp_conda_3.5_170949\conda\conda-bld\pytorch_1570813862397\work\aten\src\TH/generic/THTensorEvenMoreMath.cpp:151”

代码

result = result.scatter_(1, input.cpu(), 1)

原因

先查维度，看看target的维度是正不正常。
如果维度正常，非常大的可能是数值问题。比如label shape [32, 1, 512, 512]，对于二分类问题，想让输出尺寸 [32, 2, 512, 512]，但是报错。请检查label的数值最大值是否为1。我的代码出错原因是dataloader忘记归一化label了。