search

bubbliiiing / efficientdet-pytorch

这是一个efficientdet-pytorch的源码，可以用于训练自己的模型。
MIT License
309 stars 61 forks source link

get_dr_txt.py 报错 #7

Open zhangke1997 opened 4 years ago

zhangke1997 commented 4 years ago

RuntimeError: Error(s) in loading state_dict for EfficientDetBackbone: size mismatch for classifier.header.pointwise_conv.conv.weight: copying a param with shape torch.Size([45, 112, 1, 1]) from checkpoint, the shape in current model is torch.Size([189, 112, 1, 1]). size mismatch for classifier.header.pointwise_conv.conv.bias: copying a param with shape torch.Size([45]) from checkpoint, the shape in current model is torch.Size([189]). 用的phi=2 pth文件也改了，为什么维度对不上呢

zhangke1997 commented 4 years ago

没问题了

qingHX commented 4 years ago

I have the same problem. Can you tell me how to solve it. Thanks!!!!!!!!!!!!!

qingHX commented 4 years ago

That's My fault. The file name is wrong. Thanks!!!

buxiangqimingzia commented 4 years ago

请问怎么解决这个问题的？

n719710930 commented 3 years ago

phi and class.txt should be changed together