Skip to content

Commit 05dac9e

Browse files
committed
fix:Returning Any from function declared to return Tensor
Signed-off-by: ytl0623 <[email protected]>
1 parent 45d9877 commit 05dac9e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

monai/losses/unified_focal_loss.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -293,6 +293,6 @@ def forward(self, y_pred: torch.Tensor, y_true: torch.Tensor) -> torch.Tensor:
293293
asy_focal_loss = self.asy_focal_loss(y_pred, y_true)
294294
asy_focal_tversky_loss = self.asy_focal_tversky_loss(y_pred, y_true)
295295

296-
loss = self.weight * asy_focal_loss + (1 - self.weight) * asy_focal_tversky_loss
296+
loss: torch.Tensor = self.weight * asy_focal_loss + (1 - self.weight) * asy_focal_tversky_loss
297297

298298
return loss

0 commit comments

Comments
 (0)