From 618074b3406931a1828312e62e7c49ceea434b2e Mon Sep 17 00:00:00 2001 From: ytl0623 Date: Thu, 15 Jan 2026 16:48:48 +0800 Subject: [PATCH] Update the docstrings to match the actual code defaults Signed-off-by: ytl0623 --- monai/losses/unified_focal_loss.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) 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