-
Notifications
You must be signed in to change notification settings - Fork 1.5k
Documentation for DiceLoss is misleading #3306
Copy link
Copy link
Closed
Labels
enhancementNew feature or requestNew feature or request
Description
The documentation for DiceLoss says (my bold):
Compute average Dice loss between two tensors. It can support both multi-classes and multi-labels tasks. Input logits input (BNHW[D] where N is number of classes) is compared with ground truth target (BNHW[D]). Axis N of input is expected to have logit predictions for each class rather than being image channels, while the same axis of target can be 1 or N (one-hot format)...
The term "logit" here doesn't seem to match how DiceLoss actually works; it expects class probabilities and not logit values. That can change for one of the inputs if the parameters sigmoid or softmax are set to True, but those parameters are False by default.
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
enhancementNew feature or requestNew feature or request