Pytorch-Segmentation-multi-models icon indicating copy to clipboard operation
Pytorch-Segmentation-multi-models copied to clipboard
verified publisher icon Minerva-J

About dice_loss

Open PatrickWilliams44 opened this issue 3 years ago • 0 comments

Hi Minerva-J! My outputs.shape = [2, 3, 224, 224], targets_x.shape=[2, 224, 224], I want to use dice_loss as the loss function, it requires the input parameter shape to be [B, H, W], how should I modify it? Looking forward to your early reply! Thank you a lot.

PatrickWilliams44 avatar Aug 16 '22 00:08 PatrickWilliams44