mirror of
https://github.com/open-mmlab/mmsegmentation.git
synced 2025-06-03 22:03:48 +08:00
[Fix] Remove unexist arg in docstring #1106
This commit is contained in:
parent
3057ef6d9b
commit
ec0377328d
@ -55,8 +55,6 @@ class DiceLoss(nn.Module):
|
|||||||
Volumetric Medical Image Segmentation <https://arxiv.org/abs/1606.04797>`_.
|
Volumetric Medical Image Segmentation <https://arxiv.org/abs/1606.04797>`_.
|
||||||
|
|
||||||
Args:
|
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.
|
smooth (float): A float number to smooth loss, and avoid NaN error.
|
||||||
Default: 1
|
Default: 1
|
||||||
exponent (float): An float number to calculate denominator
|
exponent (float): An float number to calculate denominator
|
||||||
|
Loading…
x
Reference in New Issue
Block a user