mmsegmentation
mmsegmentation copied to clipboard
When combine CEloss and DiceLoss, An error occurs !
mmseg/models/decode_heads/decode_head.py
when add DiceLoss, the mismatch happens,
mmseg/models/losses/dice_loss.py
same question
TypeError: forward() got an unexpected keyword argument 'ignore_index'
I add ignore_index=255 to def forward and the bug is fixed.
How did you combine the CE loss and the Dice loss? Can you please guide me with the implementation?