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