From 2ccc7e479500589106584861f4ebf2c0a6a766f5 Mon Sep 17 00:00:00 2001 From: Kai Chen <chenkaidev@gmail.com> Date: Sun, 17 Feb 2019 15:03:04 +0800 Subject: [PATCH] minor fix for dcn check --- mmdet/models/backbones/resnet.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/mmdet/models/backbones/resnet.py b/mmdet/models/backbones/resnet.py index 80abd26..47cccfc 100644 --- a/mmdet/models/backbones/resnet.py +++ b/mmdet/models/backbones/resnet.py @@ -331,8 +331,7 @@ class ResNet(nn.Module): assert num_stages >= 1 and num_stages <= 4 self.strides = strides self.dilations = dilations - assert len(strides) == len(dilations) == len( - stage_with_dcn) == num_stages + assert len(strides) == len(dilations) == num_stages self.out_indices = out_indices assert max(out_indices) < num_stages self.style = style @@ -342,6 +341,8 @@ class ResNet(nn.Module): self.norm_eval = norm_eval self.dcn = dcn self.stage_with_dcn = stage_with_dcn + if dcn is not None: + assert len(stage_with_dcn) == num_stages self.zero_init_residual = zero_init_residual self.block, stage_blocks = self.arch_settings[depth] self.stage_blocks = stage_blocks[:num_stages] -- GitLab