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):
ctx.bufs_ = [input.new_empty(0), input.new_empty(0)] # columns, ones
cur_im2col_step = min(ctx.im2col_step, input.size(0))
assert (input.size(0) %
cur_im2col_step) == 0, 'im2col step must divide batchsize'
assert (input.size(0) % cur_im2col_step
) == 0, 'batch size must be divisible by im2col_step'
ext_module.deform_conv_forward(
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