size mismatch when loading model
#1
by
fixedot
- opened
RuntimeError: Error(s) in loading state_dict for VGG:
Missing key(s) in state_dict: "pre_logits.fc1.weight", "pre_logits.fc1.bias", "pre_logits.fc2.weight", "pre_logits.fc2.bias".
size mismatch for head.fc.weight: copying a param with shape torch.Size([100, 512]) from checkpoint, the shape in current model is torch.Size([100, 4096]).