RuntimeError in model.load_state_dict(model_dict['state_dict'],strict=False)
Jugerr opened this issue · 0 comments
Jugerr commented
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]).