pytorch， multi-target not supported at 的一种可能原因

版权声明：转载注明出处 https://blog.csdn.net/york1996/article/details/84189802

在使用交叉熵损失函数的时候，target的形状应该是和label的形状一致或者是只有batchsize这一个维度的。

如果target是这样的【batchszie，1】就会出现上述的错误。改一下试试，用squeeze（）函数降低纬度，如果不知道squeeze怎么用的，可以参考我的其他博客。pytorch下的unsqueeze和squeeze用法 - york1996的博客 - CSDN博客

这只是一种可能的原因。