bug fix for x101 gcnet (#854)

xvjiarui · Jun 22, 2019 · 9b0bcb1 · 9b0bcb1
1 parent ae856e1
commit 9b0bcb1
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 11 deletions.
diff --git a/mmdet/models/backbones/resnet.py b/mmdet/models/backbones/resnet.py
@@ -297,11 +297,11 @@ def make_res_layer(block,
     layers = []
     layers.append(
         block(
-            inplanes,
-            planes,
-            stride,
-            dilation,
-            downsample,
+            inplanes=inplanes,
+            planes=planes,
+            stride=stride,
+            dilation=dilation,
+            downsample=downsample,
             style=style,
             with_cp=with_cp,
             conv_cfg=conv_cfg,
@@ -314,10 +314,10 @@ def make_res_layer(block,
     for i in range(1, blocks):
         layers.append(
             block(
-                inplanes,
-                planes,
-                1,
-                dilation,
+                inplanes=inplanes,
+                planes=planes,
+                stride=1,
+                dilation=dilation,
                 style=style,
                 with_cp=with_cp,
                 conv_cfg=conv_cfg,

diff --git a/mmdet/models/backbones/resnext.py b/mmdet/models/backbones/resnext.py
@@ -11,12 +11,12 @@
 
 class Bottleneck(_Bottleneck):
 
-    def __init__(self, groups=1, base_width=4, *args, **kwargs):
+    def __init__(self, inplanes, planes, groups=1, base_width=4, **kwargs):
         """Bottleneck block for ResNeXt.
         If style is "pytorch", the stride-two layer is the 3x3 conv layer,
         if it is "caffe", the stride-two layer is the first 1x1 conv layer.
         """
-        super(Bottleneck, self).__init__(*args, **kwargs)
+        super(Bottleneck, self).__init__(inplanes, planes, **kwargs)
 
         if groups == 1:
             width = self.planes