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了。