mirrors/fast-reid
1
0
Fork 0
mirror of https://github.com/JDAI-CV/fast-reid.git synced 2025-06-03 14:50:47 +08:00
Code Issues Projects Releases Wiki Activity

fix R101 bottleneck missing problem

Browse Source 
Summary: add key 101 in block dict to support R101
This commit is contained in:
liaoxingyu 2020-05-26 14:48:32 +08:00
parent d4b71de3aa
commit a1cb123cfa
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
fastreid/modeling/backbones/resnet.py
View File

@ -250,7 +250,7 @@ def build_resnet_backbone(cfg):
num_blocks_per_stage = {34: [3, 4, 6, 3], 50: [3, 4, 6, 3], 101: [3, 4, 23, 3], 152: [3, 8, 36, 3], }[depth]
nl_layers_per_stage = {34: [3, 4, 6, 3], 50: [0, 2, 3, 0], 101: [0, 2, 9, 0]}[depth]
block = {34: BasicBlock, 50: Bottleneck}[depth]
block = {34: BasicBlock, 50: Bottleneck, 101: Bottleneck}[depth]
model = ResNet(last_stride, bn_norm, num_splits, with_ibn, with_se, with_nl, block,
num_blocks_per_stage, nl_layers_per_stage)
if pretrain: