Pytorch-Medical-Segmentation icon indicating copy to clipboard operation
Pytorch-Medical-Segmentation copied to clipboard
verified publisher icon MontaEllis

predict & target shape do not match

Open ll-gh opened this issue 2 years ago • 2 comments

How to solve this problem? input size:torch.Size([2, 2, 32, 32, 32]) target size: torch.Size([2, 64, 32, 32])

ll-gh avatar Aug 17 '23 11:08 ll-gh

how to solve this problem？

BelieferQAQ avatar Sep 14 '23 08:09 BelieferQAQ

我是改成这样子，就可以了。loss = criterion_ce(outputs, target_argmax) + criterion_dice(outputs_argmax, target_argmax)

QianLingjun avatar Nov 27 '23 03:11 QianLingjun