We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4411826 commit dd888deCopy full SHA for dd888de
monai/losses/focal_loss.py
@@ -276,7 +276,7 @@ def focal_loss_with_probs(
276
log_pt = torch.log(torch.clamp(pt, min=1e-8)) # Avoid log(0)
277
focal_factor = (1 - pt).pow(gamma) # (1 - pt)**gamma
278
279
- loss = -focal_factor * log_pt
+ loss: torch.Tensor = -focal_factor * log_pt
280
281
if alpha is not None:
282
# alpha if t==1; (1-alpha) if t==0
0 commit comments