From 32b1929ce16dda9eb62da0c70526476cd4f1d787 Mon Sep 17 00:00:00 2001
From: George Ni <nemonameless@qq.com>
Date: Wed, 11 Aug 2021 17:28:51 +0800
Subject: [PATCH] add BasicBlock dcn (#3947)

---
 ppdet/modeling/backbones/resnet.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ppdet/modeling/backbones/resnet.py b/ppdet/modeling/backbones/resnet.py
index c0f2a6b6c..d4bc878ea 100755
--- a/ppdet/modeling/backbones/resnet.py
+++ b/ppdet/modeling/backbones/resnet.py
@@ -186,7 +186,6 @@ class BasicBlock(nn.Layer):
                  dcn_v2=False,
                  std_senet=False):
         super(BasicBlock, self).__init__()
-        assert dcn_v2 is False, "Not implemented yet."
         assert groups == 1 and base_width == 64, 'BasicBlock only supports groups=1 and base_width=64'
 
         self.shortcut = shortcut
@@ -239,7 +238,8 @@ class BasicBlock(nn.Layer):
             norm_type=norm_type,
             norm_decay=norm_decay,
             freeze_norm=freeze_norm,
-            lr=lr)
+            lr=lr,
+            dcn_v2=dcn_v2)
 
         self.std_senet = std_senet
         if self.std_senet:
-- 
GitLab