From ec0377328d5b05a3bbad7a3b5275c8c9c5a2c1a3 Mon Sep 17 00:00:00 2001 From: KangBK0120 Date: Mon, 6 Dec 2021 21:03:25 +0900 Subject: [PATCH] [Fix] Remove unexist arg in docstring #1106 --- mmseg/models/losses/dice_loss.py | 2 -- 1 file changed, 2 deletions(-) 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