diff --git a/monai/losses/dice.py b/monai/losses/dice.py index 948749606b..2d07b17aa2 100644 --- a/monai/losses/dice.py +++ b/monai/losses/dice.py @@ -27,7 +27,7 @@ from monai.networks import one_hot from monai.utils import DiceCEReduction, LossReduction, Weight, look_up_option - +# TODO: Implement ignore_index support (#8734) class DiceLoss(_Loss): """ Compute average Dice loss between two tensors. It can support both multi-classes and multi-labels tasks.