RuntimeError: The size of tensor a (1310720) must match the size of tensor b (327680) at non-singlet

该错误表明，在进行某个操作时，张量a和b在1个非单列维上的大小不一致。

这个错误主要源于张量 input 和 target 在维度上的不匹配。具体来说，input 的大小是 1310720，而 target 的大小是 327680，它们在非单例维度1上的尺寸不相符。因此在乘法操作时，发生了维度冲突。

可能的原因：

图像大小未归一化：

• 你在代码中用 masks/255 对 masks 进行了归一化操作。检查你的 preds 和 masks 的尺寸是否匹配。在输入神经网络进行前，确保 preds 和 masks 的形状是相同的。

解决方案：

• 确认你的 preds 和 masks 都经过相同的预处理，尤其是确保它们的尺寸保持一致。可以在 train.py 中打印 preds 和 masks 的形状来进行调试：

• print(preds.shape)
  print(masks.shape)

• 将masks的大小调整与preds一致

  masks = torch.nn.functional.interpolate(masks, size=preds.shape[-2:], mode='nearest')