pytorch：计算损失函数，用于多标签文本分类

Question

我将pytorch与变压器配合使用，以使用经过训练的bert模型。而我正在计算损失函数。我收到有关不同尺寸的错误消息，这使我离开了一段时间。这是整个Notebook。这是错误：

ValueError: Target size (torch.Size([32, 14])) must be the same as input size (torch.Size([24576, 14]))

如果有任何疑问，请确保滚动浏览笔记本电脑。