thijsgelton
commented
2 years ago - Added an ignore_label parameter to the weighted dice and cross entropy loss.
- Moved the ignore_label for RobustCrossEntropyLoss to ignore_index, since ignore_label was not computing the mean afterwards (different behaviour when using ignore_label, e.g. loss was 160 and overruled the dice loss)