Fix/issue 1198 value error invalid type torch.mps.float tensor (#1199)

* fixing type and device casting

* updating version with fix

* Revert "fixing type and device casting"

This reverts commit 62682f0.

* Revert "updating version with fix"

This reverts commit 0b431b0.

* fixing type and device casting

* updating version with fix

* Update segmentation_models_pytorch/__version__.py

---------

Co-authored-by: Lukas Sanner <lukas.sanner@lgln.niedersachsen.de>
Co-authored-by: Pavel Iakubovskii <qubvel@gmail.com>

Author: 3 people

segmentation_models_pytorch/__version__.py

@@ -1 +1 @@
-__version__ = "0.5.0"
+__version__ = "0.5.1.dev0"

segmentation_models_pytorch/losses/_functional.py

@@ -66,7 +66,7 @@ def focal_loss_with_logits(
     References:
         https://github.com/open-mmlab/mmdetection/blob/master/mmdet/core/loss/losses.py
     """
-    target = target.type(output.type())
+    target = target.to(dtype=output.dtype, device=output.device)
 
     logpt = F.binary_cross_entropy_with_logits(output, target, reduction="none")
     pt = torch.exp(-logpt)