diff --git a/mmseg/models/losses/dice_loss.py b/mmseg/models/losses/dice_loss.py index 774bd1aea..79a3abfc2 100644 --- a/mmseg/models/losses/dice_loss.py +++ b/mmseg/models/losses/dice_loss.py @@ -55,8 +55,6 @@ class DiceLoss(nn.Module): Volumetric Medical Image Segmentation `_. Args: - loss_type (str, optional): Binary or multi-class loss. - Default: 'multi_class'. Options are "binary" and "multi_class". smooth (float): A float number to smooth loss, and avoid NaN error. Default: 1 exponent (float): An float number to calculate denominator