Skip to content

Documentation for DiceLoss is misleading #3306

@ebrahimebrahim

Description

@ebrahimebrahim

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.

Metadata

Metadata

Assignees

Labels

enhancementNew feature or request

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions