Skip to content

Commit bd5d850

Browse files
authored
1 parent 133375e commit bd5d850

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

ppdet/modeling/backbones/resnet.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -590,8 +590,9 @@ def __init__(self, depth=50):
590590
if depth < 50:
591591
feat_in = 256
592592
na = NameAdapter(self)
593+
block = BottleNeck if depth >= 50 else BasicBlock
593594
self.res5 = Blocks(
594-
depth, feat_in, feat_out, count=3, name_adapter=na, stage_num=5)
595+
block, feat_in, feat_out, count=3, name_adapter=na, stage_num=5)
595596
self.feat_out = feat_out if depth < 50 else feat_out * 4
596597

597598
@property

0 commit comments

Comments
 (0)