From 0a375614ca2eaf24dc0210ab173e3ac4386f22d1 Mon Sep 17 00:00:00 2001
From: ddonatien <35450772+ddonatien@users.noreply.github.com>
Date: Fri, 23 Jul 2021 21:08:58 +0800
Subject: [PATCH] Prevent kernal_normalizer to change mask dtype (#1210)

---
 mmcv/ops/carafe.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mmcv/ops/carafe.py b/mmcv/ops/carafe.py
index 4ec679189..163458572 100644
--- a/mmcv/ops/carafe.py
+++ b/mmcv/ops/carafe.py
@@ -268,7 +268,7 @@ class CARAFEPack(nn.Module):
         mask_channel = int(mask_c / float(self.up_kernel**2))
         mask = mask.view(n, mask_channel, -1, h, w)
 
-        mask = F.softmax(mask, dim=2)
+        mask = F.softmax(mask, dim=2, dtype=mask.dtype)
         mask = mask.view(n, mask_c, h, w).contiguous()
 
         return mask