-
Notifications
You must be signed in to change notification settings - Fork 7.2k
New Feature: Dice Loss #6435
Copy link
Copy link
Open
Labels
Description
🚀 The feature
Followup to #6323
Addition of Dice Loss to torchvision.
Motivation, pitch
Mainly Dice loss is used for semantic segmentation.
I want to understand the technical aspects of adding it to torchvision. Are we going to support boolean tensors or outputs from the semantic segmentation models?
Few references.
pytorch/pytorch#1249 (comment)
MONOAI
Alternatives
No response
Additional context
No response
Reactions are currently unavailable