Skip to content

Commit 9e18d85

Browse files
committed
[ADD] soft_label in DiceCELoss documentation
1 parent cae7c72 commit 9e18d85

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

monai/losses/dice.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -716,6 +716,8 @@ def __init__(
716716
label_smoothing: a value in [0, 1] range. If > 0, the labels are smoothed
717717
by the given factor to reduce overfitting.
718718
Defaults to 0.0.
719+
soft_label: whether the target contains non-binary values (soft labels) or not.
720+
If True a soft label formulation of the DiceLoss will be used.
719721
720722
"""
721723
super().__init__()

0 commit comments

Comments
 (0)