我在 Pytorch 上遇到错误，尤其是在 nllloss 上

Question

我想应用标准，其中 criterion = nn.NLLLoss() 我将它应用于输出和标签

loss = criterion(output.view(-1,1), labels.long())

地点：

*标签的形状

labels
tensor([ 1,  4,  1,  1,  4,  1,  2,  3,  2,  4,  2,  3,  3,  4,
         0,  4])
output
tensor([ 0.1829,  0.1959,  0.1909,  0.1895,  0.1914,  0.1883,  0.1895,
         0.1884,  0.1865,  0.1931,  0.1883,  0.1917,  0.1942,  0.1937,
         0.1897,  0.1934])

输出的形状 torch.Size([16])

在下面一行：

loss = criterion(output.view(-1,1), labels.long())

我收到此错误：

错误是：

RuntimeError: 断言 `cur_target >= 0 && cur_target

有什么想法吗？

Answer 1

您的标签和输出形状必须分别为 [batch_size] 和 [batch_size, n_classes]。