fix res5 (PaddlePaddle#2631)

heavengate · Apr 15, 2021 · bd5d850 · bd5d850
1 parent 133375e
commit bd5d850
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ppdet/modeling/backbones/resnet.py b/ppdet/modeling/backbones/resnet.py
@@ -590,8 +590,9 @@ def __init__(self, depth=50):
         if depth < 50:
             feat_in = 256
         na = NameAdapter(self)
+        block = BottleNeck if depth >= 50 else BasicBlock
         self.res5 = Blocks(
-            depth, feat_in, feat_out, count=3, name_adapter=na, stage_num=5)
+            block, feat_in, feat_out, count=3, name_adapter=na, stage_num=5)
         self.feat_out = feat_out if depth < 50 else feat_out * 4
 
     @property