When I want to use mobilenetX0.25_pretrain.tar as a backbone, I got the following error. What did I do wrong?

biubug6 / Pytorch_Retinaface

Retinaface get 80.99% in widerface hard val using mobilenet0.25.

MIT License

2.63k stars 774 forks source link

When I want to use mobilenetX0.25_pretrain.tar as a backbone, I got the following error. What did I do wrong? #167

Open yeahyeahye opened 3 years ago

yeahyeahye commented 3 years ago

RuntimeError: Error(s) in loading state_dict for MobileNetV1: size mismatch for stage1.0.0.weight: copying a param with shape torch.Size([8, 3, 3, 3]) from checkpoint, the shape in current model is torch.Size([8, 3, 2, 2]).