小编lec*_*ose的帖子

我是 Pytorch 的新手，我正在尝试实现一个简单的 CNN 来识别 MNIST 图像。

我正在使用 MSE 损失作为损失函数和 SGD 作为优化器来训练网络。当我参加培训时，它给了我以下内容

warning: " UserWarning: 使用与输入大小 (torch.Size([64, 10])) 不同的目标大小 (torch.Size([64]))。这可能会因广播而导致错误的结果。请确保它们的尺寸相同。”

然后我得到以下内容

error: "RuntimeError: The size of tensor a (10) must match the size of tensor b
       (64) at non-singleton dimension 1".

我尝试使用在其他问题中找到的一些解决方案来解决它，但似乎没有任何效果。这是我如何加载数据集的代码：

transform = transforms.Compose([transforms.ToTensor(), transforms.Normalize((0.5,),(0.5,))])

trainset = torchvision.datasets.MNIST(root='./data', train = True, transform = transform, download = True)
trainloader = torch.utils.data.DataLoader(trainset, batch_size = 64, shuffle = True)

testset = torchvision.datasets.MNIST(root='./data', train = False, transform = transform, download = True)
testloader …