版权声明:转载注明出处 https://blog.csdn.net/york1996/article/details/84189802
在使用交叉熵损失函数的时候,target的形状应该是和label的形状一致或者是只有batchsize这一个维度的。
如果target是这样的【batchszie,1】就会出现上述的错误。改一下试试,用squeeze()函数降低纬度,如果不知道squeeze怎么用的,可以参考我的其他博客。pytorch下的unsqueeze和squeeze用法 - york1996的博客 - CSDN博客
这只是一种可能的原因。