[Fix] Fix DCN fp16

open-mmlab · May 8, 2021 · 8ee2f20 · 8ee2f20
1 parent db6b054
commit 8ee2f20
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mmcv/ops/deform_conv.py b/mmcv/ops/deform_conv.py
@@ -70,6 +70,7 @@ def forward(ctx,
         ctx.deform_groups = deform_groups
         ctx.im2col_step = im2col_step
 
+        weight = weight.type_as(input)
         ctx.save_for_backward(input, offset, weight)
 
         output = input.new_empty(