Merge pull request #7 from AdeelH/AdeelH-patch-1

Ensure forward() always returns a tensor

Author: AdeelH

focal_loss.py

@@ -66,7 +66,7 @@ def forward(self, x: Tensor, y: Tensor) -> Tensor:
         unignored_mask = y != self.ignore_index
         y = y[unignored_mask]
         if len(y) == 0:
-            return 0.
+            return torch.tensor(0.)
         x = x[unignored_mask]
 
         # compute weighted cross entropy term: -alpha * log(pt)