Open suhuaqiang opened 2 years ago
Thank you very much for the code you provided. Could you please tell me how to solve this problem in my mailbox "test"
Traceback (most recent call last): File "test.py", line 52, in model.load_state_dict(ckpt['net']) File "/home/ubuntu/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1406, in load_state_dict raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format( RuntimeError: Error(s) in loading state_dict for DataParallel: Missing key(s) in state_dict: "module.encoder_stage1.0.weight", "module.encoder_stage1.0.bias", "module.encoder_stage1.1.weight", "module.encoder_stage1.2.weight", "module.encoder_stage1.2.bias", "module.encoder_stage1.3.weight", "module.encoder_stage2.0.weight", "module.encoder_stage2.0.bias", "module.encoder_stage2.1.weight", "module.encoder_stage2.2.weight", "module.encoder_stage2.2.bias", "module.encoder_stage2.3.weight", "module.encoder_stage2.4.weight", "module.encoder_stage2.4.bias", "module.encoder_stage2.5.weight", "module.encoder_stage3.0.weight", "module.encoder_stage3.0.bias", "module.encoder_stage3.1.weight", "module.encoder_stage3.2.weight", "module.encoder_stage3.2.bias", "module.encoder_stage3.3.weight", "module.encoder_stage3.4.weight", "module.encoder_stage3.4.bias", "module.encoder_stage3.5.weight", "module.encoder_stage4.0.weight", "module.encoder_stage4.0.bias", "module.encoder_stage4.1.weight", "module.encoder_stage4.2.weight", "module.encoder_stage4.2.bias", "module.encoder_stage4.3.weight", "module.encoder_stage4.4.weight", "module.encoder_stage4.4.bias", "module.encoder_stage4.5.weight", "module.decoder_stage1.0.weight", "module.decoder_stage1.0.bias", "module.decoder_stage1.1.weight", "module.decoder_stage1.2.weight", "module.decoder_stage1.2.bias", "module.decoder_stage1.3.weight", "module.decoder_stage1.4.weight", "module.decoder_stage1.4.bias", "module.decoder_stage1.5.weight", "module.decoder_stage2.0.weight", "module.decoder_stage2.0.bias", "module.decoder_stage2.1.weight", "module.decoder_stage2.2.weight", "module.decoder_stage2.2.bias", "module.decoder_stage2.3.weight", "module.decoder_stage2.4.weight", "module.decoder_stage2.4.bias", "module.decoder_stage2.5.weight", "module.decoder_stage3.0.weight", "module.decoder_stage3.0.bias", "module.decoder_stage3.1.weight", "module.decoder_stage3.2.weight", "module.decoder_stage3.2.bias", "module.decoder_stage3.3.weight", "module.decoder_stage3.4.weight", "module.decoder_stage3.4.bias", "module.decoder_stage3.5.weight", "module.decoder_stage4.0.weight", "module.decoder_stage4.0.bias", "module.decoder_stage4.1.weight", "module.decoder_stage4.2.weight", "module.decoder_stage4.2.bias", "module.decoder_stage4.3.weight", "module.down_conv1.0.weight", "module.down_conv1.0.bias", "module.down_conv1.1.weight", "module.down_conv2.0.weight", "module.down_conv2.0.bias", "module.down_conv2.1.weight", "module.down_conv3.0.weight", "module.down_conv3.0.bias", "module.down_conv3.1.weight", "module.down_conv4.0.weight", "module.down_conv4.0.bias", "module.down_conv4.1.weight", "module.up_conv2.0.weight", "module.up_conv2.0.bias", "module.up_conv2.1.weight", "module.up_conv3.0.weight", "module.up_conv3.0.bias", "module.up_conv3.1.weight", "module.up_conv4.0.weight", "module.up_conv4.0.bias", "module.up_conv4.1.weight". Unexpected key(s) in state_dict: "module.encoder1.weight", "module.encoder1.bias", "module.encoder2.weight", "module.encoder2.bias", "module.encoder3.weight", "module.encoder3.bias", "module.encoder4.weight", "module.encoder4.bias", "module.decoder2.weight", "module.decoder2.bias", "module.decoder3.weight", "module.decoder3.bias", "module.decoder4.weight", "module.decoder4.bias", "module.decoder5.weight", "module.decoder5.bias". size mismatch for module.map4.0.weight: copying a param with shape torch.Size([2, 2, 1, 1, 1]) from checkpoint, the shape in current model is torch.Size([2, 32, 1, 1, 1]).
Thank you very much for the code you provided. Could you please tell me how to solve this problem in my mailbox "test"
Traceback (most recent call last): File "test.py", line 52, in
model.load_state_dict(ckpt['net'])
File "/home/ubuntu/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1406, in load_state_dict
raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for DataParallel:
Missing key(s) in state_dict: "module.encoder_stage1.0.weight", "module.encoder_stage1.0.bias", "module.encoder_stage1.1.weight", "module.encoder_stage1.2.weight", "module.encoder_stage1.2.bias", "module.encoder_stage1.3.weight", "module.encoder_stage2.0.weight", "module.encoder_stage2.0.bias", "module.encoder_stage2.1.weight", "module.encoder_stage2.2.weight", "module.encoder_stage2.2.bias", "module.encoder_stage2.3.weight", "module.encoder_stage2.4.weight", "module.encoder_stage2.4.bias", "module.encoder_stage2.5.weight", "module.encoder_stage3.0.weight", "module.encoder_stage3.0.bias", "module.encoder_stage3.1.weight", "module.encoder_stage3.2.weight", "module.encoder_stage3.2.bias", "module.encoder_stage3.3.weight", "module.encoder_stage3.4.weight", "module.encoder_stage3.4.bias", "module.encoder_stage3.5.weight", "module.encoder_stage4.0.weight", "module.encoder_stage4.0.bias", "module.encoder_stage4.1.weight", "module.encoder_stage4.2.weight", "module.encoder_stage4.2.bias", "module.encoder_stage4.3.weight", "module.encoder_stage4.4.weight", "module.encoder_stage4.4.bias", "module.encoder_stage4.5.weight", "module.decoder_stage1.0.weight", "module.decoder_stage1.0.bias", "module.decoder_stage1.1.weight", "module.decoder_stage1.2.weight", "module.decoder_stage1.2.bias", "module.decoder_stage1.3.weight", "module.decoder_stage1.4.weight", "module.decoder_stage1.4.bias", "module.decoder_stage1.5.weight", "module.decoder_stage2.0.weight", "module.decoder_stage2.0.bias", "module.decoder_stage2.1.weight", "module.decoder_stage2.2.weight", "module.decoder_stage2.2.bias", "module.decoder_stage2.3.weight", "module.decoder_stage2.4.weight", "module.decoder_stage2.4.bias", "module.decoder_stage2.5.weight", "module.decoder_stage3.0.weight", "module.decoder_stage3.0.bias", "module.decoder_stage3.1.weight", "module.decoder_stage3.2.weight", "module.decoder_stage3.2.bias", "module.decoder_stage3.3.weight", "module.decoder_stage3.4.weight", "module.decoder_stage3.4.bias", "module.decoder_stage3.5.weight", "module.decoder_stage4.0.weight", "module.decoder_stage4.0.bias", "module.decoder_stage4.1.weight", "module.decoder_stage4.2.weight", "module.decoder_stage4.2.bias", "module.decoder_stage4.3.weight", "module.down_conv1.0.weight", "module.down_conv1.0.bias", "module.down_conv1.1.weight", "module.down_conv2.0.weight", "module.down_conv2.0.bias", "module.down_conv2.1.weight", "module.down_conv3.0.weight", "module.down_conv3.0.bias", "module.down_conv3.1.weight", "module.down_conv4.0.weight", "module.down_conv4.0.bias", "module.down_conv4.1.weight", "module.up_conv2.0.weight", "module.up_conv2.0.bias", "module.up_conv2.1.weight", "module.up_conv3.0.weight", "module.up_conv3.0.bias", "module.up_conv3.1.weight", "module.up_conv4.0.weight", "module.up_conv4.0.bias", "module.up_conv4.1.weight".
Unexpected key(s) in state_dict: "module.encoder1.weight", "module.encoder1.bias", "module.encoder2.weight", "module.encoder2.bias", "module.encoder3.weight", "module.encoder3.bias", "module.encoder4.weight", "module.encoder4.bias", "module.decoder2.weight", "module.decoder2.bias", "module.decoder3.weight", "module.decoder3.bias", "module.decoder4.weight", "module.decoder4.bias", "module.decoder5.weight", "module.decoder5.bias".
size mismatch for module.map4.0.weight: copying a param with shape torch.Size([2, 2, 1, 1, 1]) from checkpoint, the shape in current model is torch.Size([2, 32, 1, 1, 1]).