bubbliiiing/retinanet-pytorch

关于预训练模型

Ann331 opened this issue · 2 comments

您好,请问有resnet34的预训练模型吗?为什么我使用在其它地方找到的resnet34的预训练模型对网络进行训练,会出现如下报错信息呢:
pretrained_dict = {k: v for k, v in pretrained_dict.items() if np.shape(model_dict[k]) == np.shape(v)}
KeyError: 'conv1.weight'

emm从主干开始就行

emm从主干开始就行

好的,已经解决了,感谢您的回复。