RuntimeError in model.load_state_dict(model_dict['state_dict'],strict=False)

SoftwareGift / FeatherNets_Face-Anti-spoofing-Attack-Detection-Challenge-CVPR2019

Code for 3rd Place Solution in Face Anti-spoofing Attack Detection Challenge @ CVPR2019,model only 0.35M!!! 1.88ms(CPU)

Other

928 stars 284 forks source link

RuntimeError in model.load_state_dict(model_dict['state_dict'],strict=False) #71

Open Jugerr opened 4 years ago

Jugerr commented 4 years ago

RuntimeError: Error(s) in loading state_dict for DataParallel: size mismatch for module.fish.fish.9.4.1.weight: copying a param with shape torch.Size([1056]) from checkpoint, the shape in current model is torch.Size([1000, 1056, 1, 1]). size mismatch for module.fish.fish.9.4.1.bias: copying a param with shape torch.Size([1056]) from checkpoint, the shape in current model is torch.Size([1000]).