Skip to content

Commit dddeb6f

Browse files
committed
Fix regex pattern for resent weights names in Detectron
This should solve the performance degradation on training res101 models
1 parent 9defe95 commit dddeb6f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

lib/utils/detectron_weight_helper.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ def load_detectron_weight(net, detectron_weight_file):
2222

2323

2424
def resnet_weights_name_pattern():
25-
pattern = re.compile(r"conv1_w|conv1_gn_[sb]|res_conv1_.+|res\d_\d_.+")
25+
pattern = re.compile(r"conv1_w|conv1_gn_[sb]|res_conv1_.+|res\d+_\d+_.+")
2626
return pattern
2727

2828

0 commit comments

Comments
 (0)