ValueError: Target size (torch.Size([1, 1, 256, 256, 3])) must be the same as input size (torch.Size([1, 1, 256, 256]))
Traceback (most recent call last):
File "/home/ltt/projectUNet/tools/unet_medic/Linear_lesion_Code/UNet/train.py", line 269, in <module>
main(mode='train',args=args,writer=writer,k_fold=int(i+1))
File "/home/ltt/projectUNet/tools/unet_medic/Linear_lesion_Code/UNet/train.py", line