python - pytoch: RuntimeError: shape '[-1, 14]' 对于大小为 491520 的输入无效

Question

我正在尝试在这里学习本教程：但是，当我尝试训练我的模型时。我得到这个。

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。

非常感谢任何帮助。

标签： pythonpytorch

Answer 1

您会收到此错误，因为 491520 不是 14 的倍数。

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