fix dcn in BasicBlock

2cdc7545 · pangjm · 5305a459 · 2cdc7545
Commit 2cdc7545 authored Feb 17, 2019 by pangjm
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

mmdet/models/backbones/resnet.py mmdet/models/backbones/resnet.py +3 -1

No files found.
--- a/mmdet/models/backbones/resnet.py
+++ b/mmdet/models/backbones/resnet.py
@@ -34,8 +34,10 @@ class BasicBlock(nn.Module):
                 downsample=None,
                 style='pytorch',
                 with_cp=False,
-                 normalize=dict(type='BN')):
+                 normalize=dict(type='BN'),
+                 dcn=None):
        super(BasicBlock, self).__init__()
+        assert dcn is None, "Not implemented yet."

        self.norm1_name, norm1 = build_norm_layer(normalize, planes, postfix=1)
        self.norm2_name, norm2 = build_norm_layer(normalize, planes, postfix=2)