[quote=“XiaoAHeng, post:9, topic:17416, full:true”]
ground truth: torch.Size([1, 256, 256]); predicted truth: torch.Size([1, 2, 256, 256])
I have tarnsformed the ground truth into LongTensor by GT = GT.to(self.device, dtype=torch.long). I really don’t know what went wrong