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
ModelZoo
SOLOv2-pytorch
Commits
c17fa9a1
Commit
c17fa9a1
authored
Jan 31, 2019
by
yhcao6
Browse files
only apply dcn to layer 2, 3, 4
parent
4a4b10fb
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
1 deletion
+2
-1
mmdet/models/backbones/resnext.py
mmdet/models/backbones/resnext.py
+2
-1
No files found.
mmdet/models/backbones/resnext.py
View file @
c17fa9a1
...
...
@@ -183,6 +183,7 @@ class ResNeXt(ResNet):
for
i
,
num_blocks
in
enumerate
(
self
.
stage_blocks
):
stride
=
self
.
strides
[
i
]
dilation
=
self
.
dilations
[
i
]
dcn
=
self
.
dcn
if
self
.
stage_with_dcn
[
i
]
else
None
planes
=
64
*
2
**
i
res_layer
=
make_res_layer
(
self
.
block
,
...
...
@@ -196,7 +197,7 @@ class ResNeXt(ResNet):
style
=
self
.
style
,
with_cp
=
self
.
with_cp
,
normalize
=
self
.
normalize
,
dcn
=
self
.
dcn
)
dcn
=
dcn
)
self
.
inplanes
=
planes
*
self
.
block
.
expansion
layer_name
=
'layer{}'
.
format
(
i
+
1
)
self
.
add_module
(
layer_name
,
res_layer
)
...
...
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