Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
MMCV
Commits
71341e9a
Unverified
Commit
71341e9a
authored
Nov 23, 2021
by
Jiazhen Wang
Committed by
GitHub
Nov 23, 2021
Browse files
[Fix] fix assert (#1490)
parent
78305e78
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
mmcv/ops/deform_conv.py
mmcv/ops/deform_conv.py
+2
-2
No files found.
mmcv/ops/deform_conv.py
View file @
71341e9a
...
...
@@ -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 divi
de batchsize
'
assert
(
input
.
size
(
0
)
%
cur_im2col_step
)
==
0
,
'batch size
must
be
divi
sible by im2col_step
'
ext_module
.
deform_conv_forward
(
input
,
weight
,
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment