标签 > dimension out of range [-1

这个报错是图像分割 UNet 训练里高频坑，根源是CrossEntropyLoss 对输入、标签的维度顺序要求固定：UNet 输出是 [B, C, H, W]（bat