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

时间:2024-04-03 12:01:16

@[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])

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

问题解决

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

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

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


2020.3.5
希望能帮到大家。