semseg icon indicating copy to clipboard operation
semseg copied to clipboard

the pretrained resnet50 cannot be loaded

Open linshuxi opened this issue 4 years ago • 1 comments

Hi, the resnet50 pretrained model from "https://download.pytorch.org/models/resnet50-19c8e357.pth" cannot be loaded into the model. The error is : RuntimeError: Error(s) in loading state_dict for ResNet: Missing key(s) in state_dict: "conv2.weight", "bn2.weight", "bn2.bias", "bn2.running_mean", "bn2.running_var", "conv3.weight", "bn3.weight", "bn3.bias", "bn3.running_mean", "bn3.running_var". size mismatch for conv1.weight: copying a param with shape torch.Size([64, 3, 7, 7]) from checkpoint, the shape in current model is torch.Size([64, 3, 3, 3]). size mismatch for layer1.0.conv1.weight: copying a param with shape torch.Size([64, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([64, 128, 1, 1]). size mismatch for layer1.0.downsample.0.weight: copying a param with shape torch.Size([256, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([256, 128, 1,

linshuxi avatar Jul 31 '20 02:07 linshuxi

Do you have resloved this question? I ran into the same problem.

changccanddd avatar Oct 10 '20 08:10 changccanddd