When combine CEloss and DiceLoss, An error occurs !

[wuji3]

mmseg/models/decode_heads/decode_head.py

when add DiceLoss, the mismatch happens,
mmseg/models/losses/dice_loss.py

[caj-github]

same question

[wch313]

TypeError: forward() got an unexpected keyword argument 'ignore_index'

[jiady1990]

I add ignore_index=255 to def forward and the bug is fixed.

[Masrur02]

How did you combine the CE loss and the Dice loss? Can you please guide me with the implementation?

[xhlho]

I added ignore_index=255 to def forward, and the error was fixed. However, I encountered a new problem while using it: Runtime Error: The size of tensor a (3686400) must match the size of tensor b (1228800) at non single dimension 1. Can someone help me solve it?