diff --git a/monai/losses/unified_focal_loss.py b/monai/losses/unified_focal_loss.py index 06704c0104..b58b406a7e 100644 --- a/monai/losses/unified_focal_loss.py +++ b/monai/losses/unified_focal_loss.py @@ -108,7 +108,7 @@ def __init__( Args: to_onehot_y : whether to convert `y` into the one-hot format. Defaults to False. delta : weight of the background. Defaults to 0.7. - gamma : value of the exponent gamma in the definition of the Focal loss . Defaults to 0.75. + gamma : value of the exponent gamma in the definition of the Focal loss . Defaults to 2. epsilon : it defines a very small number each time. simmily smooth value. Defaults to 1e-7. """ super().__init__(reduction=LossReduction(reduction).value) @@ -168,9 +168,8 @@ def __init__( to_onehot_y : whether to convert `y` into the one-hot format. Defaults to False. num_classes : number of classes, it only supports 2 now. Defaults to 2. delta : weight of the background. Defaults to 0.7. - gamma : value of the exponent gamma in the definition of the Focal loss. Defaults to 0.75. - epsilon : it defines a very small number each time. simmily smooth value. Defaults to 1e-7. - weight : weight for each loss function, if it's none it's 0.5. Defaults to None. + gamma : value of the exponent gamma in the definition of the Focal loss. Defaults to 0.5. + weight : weight for each loss function. Defaults to 0.5. Example: >>> import torch