Fix Accuracy metric (qubvel#186)

wouterzwerink · Apr 21, 2020 · a48886c · a48886c
1 parent ecc848f
commit a48886c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/segmentation_models_pytorch/utils/functional.py b/segmentation_models_pytorch/utils/functional.py
@@ -77,7 +77,7 @@ def accuracy(pr, gt, threshold=0.5, ignore_channels=None):
     pr = _threshold(pr, threshold=threshold)
     pr, gt = _take_channels(pr, gt, ignore_channels=ignore_channels)
 
-    tp = torch.sum(gt == pr)
+    tp = torch.sum(gt == pr, dtype=pr.dtype)
     score = tp / gt.view(-1).shape[0]
     return score