"Please ensure they have the same size.".format(target.size(), input.size()) ValueError: Using a target size (torch.Size([64, 35])) that is different to the input size (torch.Size([16, 35])) is deprecated. Please ensure they have the same size.
时间: 2024-03-04 12:54:12 浏览: 114
这个错误是由PyTorch中的损失函数产生的。错误的原因是输入的目标张量(target tensor)和模型输出的张量(input tensor)的形状(size)不匹配,即目标张量的形状与模型输出的形状不同。在PyTorch中,大多数损失函数都要求输入的目标张量和模型输出张量的形状相同。
在这个错误信息中,目标张量的形状是[64, 35],而模型输出的形状是[16, 35]。这意味着目标张量的第一个维度是输入张量的四倍,而第二个维度相同。因此,您需要检查目标张量和模型输出张量的形状是否正确,并使用适当的方法调整它们的形状,使其匹配。如果您确定目标张量和模型输出张量的形状应该不同,您可以考虑使用适当的损失函数,例如CrossEntropyLoss,它可以处理输入形状不同的情况。
阅读全文
相关推荐







