pytorch ValueError：目标和输入必须具有相同的数字

Question

我尝试通过此链接进行教程，但无法解决一个问题。

part 1 part2

第2部分中的问题

model.eval()
y_pred = model(test_input)
before_train = criterion(y_pred.squeeze(), test_output)

Output:
ValueError: Target and input must have the same number of elements. target nelement (60) != input nelement (20)

print（y_pred.squeeze（））

  tensor([0.5050, 0.5063, 0.5000, 0.5040, 0.5050, 0.4984, 0.5041, 0.4939, 0.5016,
    0.5033, 0.4993, 0.5015, 0.5058, 0.5016, 0.5009, 0.4968, 0.5015, 0.5022,
    0.4953, 0.5050], grad_fn=<SqueezeBackward0>)

print（test_output）

    tensor([[0., 1., 0.],
    [0., 1., 0.],
    [0., 1., 0.],
    [1., 0., 0.],
    [0., 0., 1.],
    [1., 0., 0.],
    [0., 0., 1.],
    [1., 0., 0.],
    [0., 0., 1.],
    [0., 1., 0.],
    [1., 0., 0.],
    [0., 0., 1.],
    [0., 1., 0.],
    [1., 0., 0.],
    [0., 0., 1.],
    [1., 0., 0.],
    [0., 0., 1.],
    [0., 1., 0.],
    [0., 1., 0.],
    [0., 1., 0.]])

我需要更改什么？我所有的代码和数据都与教程中的相同谢谢！