When I want to use mobilenetX0.25_pretrain.tar as a backbone, I got the following error. What did I do wrong?
yeahyeahye opened this issue · comments
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]).