RuntimeError: size mismatch (got input: [10], target: [1])
时间: 2023-12-22 11:03:08 浏览: 191
这个错误通常发生在神经网络训练时,表示输入和目标的大小不匹配。具体来说,这里的输入大小是10,而目标大小是1,因此它们不兼容。这个错误可能是由于网络的最后一层输出大小和目标大小不匹配造成的。你可以检查网络结构和目标数据,确保它们匹配,并且网络的输出大小和目标大小相同。此外,你可以检查网络的输入数据,确保它们与网络的输入层大小相同。
相关问题
return torch._C._nn.cross_entropy_loss(input, target, weight, _Reduction.get_enum(reduction), ignore_index, label_smoothing) RuntimeError: size mismatch (got input: [4096], target: [1])
这个错误通常是因为输入和目标张量的形状不一致导致的。你可以检查一下你的输入和目标张量的形状是否一致,特别是在 batch size 维度上是否相同。另外,你还可以尝试使用 `torch.nn.CrossEntropyLoss` 来代替 `torch._C._nn.cross_entropy_loss`,它是一个更高级的接口,会自动处理输入和目标张量的形状问题。
阅读全文