Unverified Commit 9b0bcb1c authored by Kai Chen's avatar Kai Chen Committed by GitHub
Browse files

bug fix for x101 gcnet (#854)

parent ae856e11
......@@ -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,
......
......@@ -11,12 +11,12 @@ from ..utils import build_conv_layer, build_norm_layer
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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment