mirrors/mmcv
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmcv.git synced 2025-06-03 21:54:52 +08:00
Code Issues Projects Releases Wiki Activity

[Fix] fix assert (#1490)

Browse Source
This commit is contained in:
Jiazhen Wang 2021-11-23 18:38:32 +08:00 committed by GitHub
parent 78305e7811
commit 71341e9a0a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mmcv/ops/deform_conv.py
View File

@ -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,