Unverified Commit 9b49fcc6 authored by Ace's avatar Ace Committed by GitHub
Browse files

Update modulated_deform_conv.py (#1605)

change bias to the same type as input
parent 690a77fa
...@@ -66,6 +66,7 @@ class ModulatedDeformConv2dFunction(Function): ...@@ -66,6 +66,7 @@ class ModulatedDeformConv2dFunction(Function):
# whatever the pytorch version is. # whatever the pytorch version is.
input = input.type_as(offset) input = input.type_as(offset)
weight = weight.type_as(input) weight = weight.type_as(input)
bias = bias.type_as(input)
ctx.save_for_backward(input, offset, mask, weight, bias) ctx.save_for_backward(input, offset, mask, weight, bias)
output = input.new_empty( output = input.new_empty(
ModulatedDeformConv2dFunction._output_size(ctx, input, weight)) ModulatedDeformConv2dFunction._output_size(ctx, input, weight))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment