Unverified Commit 71341e9a authored by Jiazhen Wang's avatar Jiazhen Wang Committed by GitHub
Browse files

[Fix] fix assert (#1490)

parent 78305e78
...@@ -87,8 +87,8 @@ class DeformConv2dFunction(Function): ...@@ -87,8 +87,8 @@ class DeformConv2dFunction(Function):
ctx.bufs_ = [input.new_empty(0), input.new_empty(0)] # columns, ones ctx.bufs_ = [input.new_empty(0), input.new_empty(0)] # columns, ones
cur_im2col_step = min(ctx.im2col_step, input.size(0)) cur_im2col_step = min(ctx.im2col_step, input.size(0))
assert (input.size(0) % assert (input.size(0) % cur_im2col_step
cur_im2col_step) == 0, 'im2col step must divide batchsize' ) == 0, 'batch size must be divisible by im2col_step'
ext_module.deform_conv_forward( ext_module.deform_conv_forward(
input, input,
weight, 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