Mr. Long
Results
2
issues of
Mr. Long
pretrained_dict = {k: v for k, v in pretrained_dict.items() if np.shape(model_dict[k]) == np.shape(v)} 使用这一语句会报错 KeyError: '_conv_stem.weight' 检查发现在主干网络部分 model_dict 的键为 ’backbone.model._conv_stem.weight‘ 而非 '_conv_stem.weight'