pytoch:RuntimeError:形状“[1，14]”对于大小为491520的输入无效

Question

我试图遵循本教程here：然而，当我试图训练我的模型时。我明白了

RuntimeError: shape '[-1, 14]' is invalid for input of size 491520

以下是导致此错误的行：

logits = outputs[0]
loss_func = BCEWithLogitsLoss() 
loss = loss_func(logits.view(-1,num_labels),b_labels.type_as(logits).view(-1,num_labels))

我有14个输出，你可以从错误中猜出来。当我检查logits的形状时，我得到了这个32 * 20 * 768

非常感谢您的帮助

Answer 1

您得到这个错误是因为491520不是14的倍数

PyTorch没有办法将其重塑为(-1, 14)。 确保logits张量的总大小是14的倍数