pytorch: RuntimeError: 1only batches of spatial targets supported but got targets of size : : xx

@[TOC](RuntimeError: 1only batches of spatial targets supported (non-empty 3D tensors) but got targets of size: : [64])

---

问题描述

今天在使用pytorch编写修改网络结构的时候，遇到了一个问题：
如下图所示，是求网络输出和label之间交叉熵的时候遇到维度不匹配。
所以输出classifierout和label的维度，分别是：
classifierou t.data.shape: torch.Size([64, 5, 1, 1])
label.data.shape: torch.Size([64])

而我原来编写网络结构可以正常运行的时候，输出分别是：
classifierou t.data.shape: torch.Size([64, 5])
label.data.shape: torch.Size([64])

问题解决

需要在网络输出的最后加上：
x.view(x.size(0), self.num_classes)
将网络输出的维度view为[64, 5]即可

可能大家报这个错误的时候，具体原因不太一样，但基本上从维度不匹配的角度去解决和修改应该问题都不大。

---

2020.3.5
希望能帮到大家。