DiceFocalLoss does not have the option to run softmax focal loss

Project-MONAI / MONAI

AI Toolkit for Healthcare Imaging

https://monai.io/

Apache License 2.0

5.93k stars 1.09k forks source link

Open ferreirafabio80 opened 4 days ago

ferreirafabio80 commented 4 days ago

The 'use_softmax' arg is missing in FocalLoss in the DiceFocalLoss class. See the screenshot.

KumoLiu commented 3 days ago

Hi @yiheng-wang-nv, I noticed you added this class previously. Do you remember why we not include the activation function for FocalLoss?