consistent param list of block

2018-10-09 21:08:03 +08:00 · 2018-10-09 21:08:03 +08:00 · 867d5a5003
parent e363982ada
commit 867d5a5003
1 changed files with 3 additions and 1 deletions
--- a/mmcv/cnn/resnet.py
+++ b/mmcv/cnn/resnet.py
@ -28,7 +28,8 @@ class BasicBlock(nn.Module):
                 stride=1,
                 dilation=1,
                 downsample=None,
-                 style='pytorch'):
+                 style='pytorch',
+                 with_cp=False):
        super(BasicBlock, self).__init__()
        self.conv1 = conv3x3(inplanes, planes, stride, dilation)
        self.bn1 = nn.BatchNorm2d(planes)
@ -38,6 +39,7 @@ class BasicBlock(nn.Module):
        self.downsample = downsample
        self.stride = stride
        self.dilation = dilation
+        assert not with_cp

    def forward(self, x):
        residual = x