未验证 提交 9f9db4b7 编写于 作者: W wangguanzhong 提交者: GitHub

fix match weight (#3280)

上级 edd7bc4d
......@@ -164,7 +164,7 @@ def match_state_dict(model_state_dict, weight_state_dict):
# In Faster RCNN, res5 pretrained weights have prefix of backbone,
# however, the corresponding model weights have difficult prefix,
# bbox_head.
b = b.strip('backbone.')
b = b[9:]
return a == b or a.endswith("." + b)
match_matrix = np.zeros([len(model_keys), len(weight_keys)])
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册