[Fix] Fix the bug when input mask is not '0-1-tensor' in masked_conv (o…

…pen-mmlab#2423)
ckirchhoff2021 · Dec 21, 2022 · 5ee553b · 5ee553b
1 parent 2e92c88
commit 5ee553b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mmcv/ops/masked_conv.py b/mmcv/ops/masked_conv.py
@@ -58,6 +58,8 @@ def forward(ctx,
             if mask.size()[1:] != output.size()[2:]:
                 raise ValueError(
                     'The mask is inconsistent with the shape of output_conv.')
+            mask = mask > 0
+            mask = mask.type(output.dtype)
             output = output * mask
             return output