pytorch导入训练好的模型时报错Missing key(s) in state_dict,Unexpected key(s) in state_dict

完整报错信息如下:

File "F:/pytorch/anti-spoofing_depth/living_test.py", line 53, in run_test
    network.load_state_dict(torch.load(initial_checkpoint, map_location=lambda storage, loc: storage))
  File "D:\anaconda\lib\site-packages\torch\nn\modules\module.py", line 845, in load_state_dict
    self.__class__.__name__, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for Net:
	Missing key(s) in state_dict: "first_bn.weight", "first_bn.bias", "first_bn.running_mean", "first_bn.running_var", "encoder.layer0.conv1.weight", "encoder.layer0.bn1.weight", "encoder.layer0.bn1.bias", "encoder.layer0.bn1.running_mean", "encoder.layer0.bn1.running_var", "encoder.layer1.0.conv1.weight", "encoder.layer1.0.bn1.weight", "encoder.layer1.0.bn1.bias", "encoder.layer1.0.bn1.running_mean", "encoder.layer1.0.bn1.running_var", "encoder.layer1.0.conv2.weight", "encoder.layer1.0.bn2.weight", "encoder.layer1.0.bn2.bias", "encoder.layer1.0.bn2.running_mean", "encoder.layer1.0.bn2.running_var", "encoder.layer1.0.conv3.weight", "encoder.layer1.0.bn3.weight", "encoder.layer1.0.bn3.bias", "encoder.layer1.0.bn3.running_mean", "encoder.layer1.0.bn3.running_var", "encoder.layer1.0.se_module.fc1.weight", "encoder.layer1.0.se_module.fc1.bias", "encoder.layer1.0.se_module.fc2.weight", "encoder.layer1.0.se_module.fc2.bias", "encoder.layer1.0.downsample.0.weight", "encoder.layer1.0.downsample.1.weight", "encoder.layer1.0.downsample.1.bias", "encoder.layer1.0.downsample.1.running_mean", "encoder.layer1.0.downsample.1.running_var", "encoder.layer1.1.conv1.weight", "encoder.layer1.1.bn1.weight", "encoder.layer1.1.bn1.bias", "encoder.layer1.1.bn1.running_mean", "encoder.layer1.1.bn1.running_var", "encoder.layer1.1.conv2.weight", "encoder.layer1.1.bn2.weight", "encoder.layer1.1.bn2.bias", "encoder.layer1.1.bn2.running_mean", "encoder.layer1.1.bn2.running_var", "encoder.layer1.1.conv3.weight", "encoder.layer1.1.bn3.weight", "encoder.layer1.1.bn3.bias", "encoder.layer1.1.bn3.running_mean", "encoder.layer1.1.bn3.running_var", "encoder.layer1.1.se_module.fc1.weight", "encoder.layer1.1.se_module.fc1.bias", "encoder.layer1.1.se_module.fc2.weight", "encoder.layer1.1.se_module.fc2.bias", "encoder.layer1.2.conv1.weight", "encoder.layer1.2.bn1.weight", "encoder.layer1.2.bn1.bias", "encoder.layer1.2.bn1.running_mean", "encoder.layer1.2.bn1.running_var", "encoder.layer1.2.conv2.weight", "encoder.layer1.2.bn2.weight", "encoder.layer1.2.bn2.bias", "encoder.layer1.2.bn2.running_mean", "encoder.layer1.2.bn2.running_var", "encoder.layer1.2.conv3.weight", "encoder.layer1.2.bn3.weight", "encoder.layer1.2.bn3.bias", "encoder.layer1.2.bn3.running_mean", "encoder.layer1.2.bn3.running_var", "encoder.layer1.2.se_module.fc1.weight", "encoder.layer1.2.se_module.fc1.bias", "encoder.layer1.2.se_module.fc2.weight", "encoder.layer1.2.se_module.fc2.bias", "encoder.layer2.0.conv1.weight", "encoder.layer2.0.bn1.weight", "encoder.layer2.0.bn1.bias", "encoder.layer2.0.bn1.running_mean", "encoder.layer2.0.bn1.running_var", "encoder.layer2.0.conv2.weight", "encoder.layer2.0.bn2.weight", "encoder.layer2.0.bn2.bias", "encoder.layer2.0.bn2.running_mean", "encoder.layer2.0.bn2.running_var", "encoder.layer2.0.conv3.weight", "encoder.layer2.0.bn3.weight", "encoder.layer2.0.bn3.bias", "encoder.layer2.0.bn3.running_mean", "encoder.layer2.0.bn3.running_var", "encoder.layer2.0.se_module.fc1.weight", "encoder.layer2.0.se_module.fc1.bias", "encoder.layer2.0.se_module.fc2.weight", "encoder.layer2.0.se_module.fc2.bias", "encoder.layer2.0.downsample.0.weight", "encoder.layer2.0.downsample.1.weight", "encoder.layer2.0.downsample.1.bias", "encoder.layer2.0.downsample.1.running_mean", "encoder.layer2.0.downsample.1.running_var", "encoder.layer2.1.conv1.weight", "encoder.layer2.1.bn1.weight", "encoder.layer2.1.bn1.bias", "encoder.layer2.1.bn1.running_mean", "encoder.layer2.1.bn1.running_var", "encoder.layer2.1.conv2.weight", "encoder.layer2.1.bn2.weight", "encoder.layer2.1.bn2.bias", "encoder.layer2.1.bn2.running_mean", "encoder.layer2.1.bn2.running_var", "encoder.layer2.1.conv3.weight", "encoder.layer2.1.bn3.weight", "encoder.layer2.1.bn3.bias", "encoder.layer2.1.bn3.running_mean", "encoder.layer2.1.bn3.running_var", "encoder.layer2.1.se_module.fc1.weight", "encoder.layer2.1.se_module.fc1.bias", "encoder.layer2.1.se_module.fc2.weight", "encoder.layer2.1.se_module.fc2.bias", "encoder.layer2.2.conv1.weight", "encoder.layer2.2.bn1.weight", "encoder.layer2.2.bn1.bias", "encoder.layer2.2.bn1.running_mean", "encoder.layer2.2.bn1.running_var", "encoder.layer2.2.conv2.weight", "encoder.layer2.2.bn2.weight", "encoder.layer2.2.bn2.bias", "encoder.layer2.2.bn2.running_mean", "encoder.layer2.2.bn2.running_var", "encoder.layer2.2.conv3.weight", "encoder.layer2.2.bn3.weight", "encoder.layer2.2.bn3.bias", "encoder.layer2.2.bn3.running_mean", "encoder.layer2.2.bn3.running_var", "encoder.layer2.2.se_module.fc1.weight", "encoder.layer2.2.se_module.fc1.bias", "encoder.layer2.2.se_module.fc2.weight", "encoder.layer2.2.se_module.fc2.bias", "encoder.layer2.3.conv1.weight", "encoder.layer2.3.bn1.weight", "encoder.layer2.3.bn1.bias", "encoder.layer2.3.bn1.running_mean", "encoder.layer2.3.bn1.running_var", "encoder.layer2.3.conv2.weight", "encoder.layer2.3.bn2.weight", "encoder.layer2.3.bn2.bias", "encoder.layer2.3.bn2.running_mean", "encoder.layer2.3.bn2.running_var", "encoder.layer2.3.conv3.weight", "encoder.layer2.3.bn3.weight", "encoder.layer2.3.bn3.bias", "encoder.layer2.3.bn3.running_mean", "encoder.layer2.3.bn3.running_var", "encoder.layer2.3.se_module.fc1.weight", "encoder.layer2.3.se_module.fc1.bias", "encoder.layer2.3.se_module.fc2.weight", "encoder.layer2.3.se_module.fc2.bias", "encoder.layer3.0.conv1.weight", "encoder.layer3.0.bn1.weight", "encoder.layer3.0.bn1.bias", "encoder.layer3.0.bn1.running_mean", "encoder.layer3.0.bn1.running_var", "encoder.layer3.0.conv2.weight", "encoder.layer3.0.bn2.weight", "encoder.layer3.0.bn2.bias", "encoder.layer3.0.bn2.running_mean", "encoder.layer3.0.bn2.running_var", "encoder.layer3.0.conv3.weight", "encoder.layer3.0.bn3.weight", "encoder.layer3.0.bn3.bias", "encoder.layer3.0.bn3.running_mean", "encoder.layer3.0.bn3.running_var", "encoder.layer3.0.se_module.fc1.weight", "encoder.layer3.0.se_module.fc1.bias", "encoder.layer3.0.se_module.fc2.weight", "encoder.layer3.0.se_module.fc2.bias", "encoder.layer3.0.downsample.0.weight", "encoder.layer3.0.downsample.1.weight", "encoder.layer3.0.downsample.1.bias", "encoder.layer3.0.downsample.1.running_mean", "encoder.layer3.0.downsample.1.running_var", "encoder.layer3.1.conv1.weight", "encoder.layer3.1.bn1.weight", "encoder.layer3.1.bn1.bias", "encoder.layer3.1.bn1.running_mean", "encoder.layer3.1.bn1.running_var", "encoder.layer3.1.conv2.weight", "encoder.layer3.1.bn2.weight", "encoder.layer3.1.bn2.bias", "encoder.layer3.1.bn2.running_mean", "encoder.layer3.1.bn2.running_var", "encoder.layer3.1.conv3.weight", "encoder.layer3.1.bn3.weight", "encoder.layer3.1.bn3.bias", "encoder.layer3.1.bn3.running_mean", "encoder.layer3.1.bn3.running_var", "encoder.layer3.1.se_module.fc1.weight", "encoder.layer3.1.se_module.fc1.bias", "encoder.layer3.1.se_module.fc2.weight", "encoder.layer3.1.se_module.fc2.bias", "encoder.layer3.2.conv1.weight", "encoder.layer3.2.bn1.weight", "encoder.layer3.2.bn1.bias", "encoder.layer3.2.bn1.running_mean", "encoder.layer3.2.bn1.running_var", "encoder.layer3.2.conv2.weight", "encoder.layer3.2.bn2.weight", "encoder.layer3.2.bn2.bias", "encoder.layer3.2.bn2.running_mean", "encoder.layer3.2.bn2.running_var", "encoder.layer3.2.conv3.weight", "encoder.layer3.2.bn3.weight", "encoder.layer3.2.bn3.bias", "encoder.layer3.2.bn3.running_mean", "encoder.layer3.2.bn3.running_var", "encoder.layer3.2.se_module.fc1.weight", "encoder.layer3.2.se_module.fc1.bias", "encoder.layer3.2.se_module.fc2.weight", "encoder.layer3.2.se_module.fc2.bias", "encoder.layer3.3.conv1.weight", "encoder.layer3.3.bn1.weight", "encoder.layer3.3.bn1.bias", "encoder.layer3.3.bn1.running_mean", "encoder.layer3.3.bn1.running_var", "encoder.layer3.3.conv2.weight", "encoder.layer3.3.bn2.weight", "encoder.layer3.3.bn2.bias", "encoder.layer3.3.bn2.running_mean", "encoder.layer3.3.bn2.running_var", "encoder.layer3.3.conv3.weight", "encoder.layer3.3.bn3.weight", "encoder.layer3.3.bn3.bias", "encoder.layer3.3.bn3.running_mean", "encoder.layer3.3.bn3.running_var", "encoder.layer3.3.se_module.fc1.weight", "encoder.layer3.3.se_module.fc1.bias", "encoder.layer3.3.se_module.fc2.weight", "encoder.layer3.3.se_module.fc2.bias", "encoder.layer4.0.conv1.weight", "encoder.layer4.0.bn1.weight", "encoder.layer4.0.bn1.bias", "encoder.layer4.0.bn1.running_mean", "encoder.layer4.0.bn1.running_var", "encoder.layer4.0.conv2.weight", "encoder.layer4.0.bn2.weight", "encoder.layer4.0.bn2.bias", "encoder.layer4.0.bn2.running_mean", "encoder.layer4.0.bn2.running_var", "encoder.layer4.0.conv3.weight", "encoder.layer4.0.bn3.weight", "encoder.layer4.0.bn3.bias", "encoder.layer4.0.bn3.running_mean", "encoder.layer4.0.bn3.running_var", "encoder.layer4.0.se_module.fc1.weight", "encoder.layer4.0.se_module.fc1.bias", "encoder.layer4.0.se_module.fc2.weight", "encoder.layer4.0.se_module.fc2.bias", "encoder.layer4.0.downsample.0.weight", "encoder.layer4.0.downsample.1.weight", "encoder.layer4.0.downsample.1.bias", "encoder.layer4.0.downsample.1.running_mean", "encoder.layer4.0.downsample.1.running_var", "encoder.layer4.1.conv1.weight", "encoder.layer4.1.bn1.weight", "encoder.layer4.1.bn1.bias", "encoder.layer4.1.bn1.running_mean", "encoder.layer4.1.bn1.running_var", "encoder.layer4.1.conv2.weight", "encoder.layer4.1.bn2.weight", "encoder.layer4.1.bn2.bias", "encoder.layer4.1.bn2.running_mean", "encoder.layer4.1.bn2.running_var", "encoder.layer4.1.conv3.weight", "encoder.layer4.1.bn3.weight", "encoder.layer4.1.bn3.bias", "encoder.layer4.1.bn3.running_mean", "encoder.layer4.1.bn3.running_var", "encoder.layer4.1.se_module.fc1.weight", "encoder.layer4.1.se_module.fc1.bias", "encoder.layer4.1.se_module.fc2.weight", "encoder.layer4.1.se_module.fc2.bias", "encoder.layer4.2.conv1.weight", "encoder.layer4.2.bn1.weight", "encoder.layer4.2.bn1.bias", "encoder.layer4.2.bn1.running_mean", "encoder.layer4.2.bn1.running_var", "encoder.layer4.2.conv2.weight", "encoder.layer4.2.bn2.weight", "encoder.layer4.2.bn2.bias", "encoder.layer4.2.bn2.running_mean", "encoder.layer4.2.bn2.running_var", "encoder.layer4.2.conv3.weight", "encoder.layer4.2.bn3.weight", "encoder.layer4.2.bn3.bias", "encoder.layer4.2.bn3.running_mean", "encoder.layer4.2.bn3.running_var", "encoder.layer4.2.se_module.fc1.weight", "encoder.layer4.2.se_module.fc1.bias", "encoder.layer4.2.se_module.fc2.weight", "encoder.layer4.2.se_module.fc2.bias", "encoder.last_linear.weight", "encoder.last_linear.bias", "conv1.conv1.weight", "conv1.bn1.weight", "conv1.bn1.bias", "conv1.bn1.running_mean", "conv1.bn1.running_var", "conv2.0.conv1.weight", "conv2.0.bn1.weight", "conv2.0.bn1.bias", "conv2.0.bn1.running_mean", "conv2.0.bn1.running_var", "conv2.0.conv2.weight", "conv2.0.bn2.weight", "conv2.0.bn2.bias", "conv2.0.bn2.running_mean", "conv2.0.bn2.running_var", "conv2.0.conv3.weight", "conv2.0.bn3.weight", "conv2.0.bn3.bias", "conv2.0.bn3.running_mean", "conv2.0.bn3.running_var", "conv2.0.se_module.fc1.weight", "conv2.0.se_module.fc1.bias", "conv2.0.se_module.fc2.weight", "conv2.0.se_module.fc2.bias", "conv2.0.downsample.0.weight", "conv2.0.downsample.1.weight", "conv2.0.downsample.1.bias", "conv2.0.downsample.1.running_mean", "conv2.0.downsample.1.running_var", "conv2.1.conv1.weight", "conv2.1.bn1.weight", "conv2.1.bn1.bias", "conv2.1.bn1.running_mean", "conv2.1.bn1.running_var", "conv2.1.conv2.weight", "conv2.1.bn2.weight", "conv2.1.bn2.bias", "conv2.1.bn2.running_mean", "conv2.1.bn2.running_var", "conv2.1.conv3.weight", "conv2.1.bn3.weight", "conv2.1.bn3.bias", "conv2.1.bn3.running_mean", "conv2.1.bn3.running_var", "conv2.1.se_module.fc1.weight", "conv2.1.se_module.fc1.bias", "conv2.1.se_module.fc2.weight", "conv2.1.se_module.fc2.bias", "conv2.2.conv1.weight", "conv2.2.bn1.weight", "conv2.2.bn1.bias", "conv2.2.bn1.running_mean", "conv2.2.bn1.running_var", "conv2.2.conv2.weight", "conv2.2.bn2.weight", "conv2.2.bn2.bias", "conv2.2.bn2.running_mean", "conv2.2.bn2.running_var", "conv2.2.conv3.weight", "conv2.2.bn3.weight", "conv2.2.bn3.bias", "conv2.2.bn3.running_mean", "conv2.2.bn3.running_var", "conv2.2.se_module.fc1.weight", "conv2.2.se_module.fc1.bias", "conv2.2.se_module.fc2.weight", "conv2.2.se_module.fc2.bias", "conv3.0.conv1.weight", "conv3.0.bn1.weight", "conv3.0.bn1.bias", "conv3.0.bn1.running_mean", "conv3.0.bn1.running_var", "conv3.0.conv2.weight", "conv3.0.bn2.weight", "conv3.0.bn2.bias", "conv3.0.bn2.running_mean", "conv3.0.bn2.running_var", "conv3.0.conv3.weight", "conv3.0.bn3.weight", "conv3.0.bn3.bias", "conv3.0.bn3.running_mean", "conv3.0.bn3.running_var", "conv3.0.se_module.fc1.weight", "conv3.0.se_module.fc1.bias", "conv3.0.se_module.fc2.weight", "conv3.0.se_module.fc2.bias", "conv3.0.downsample.0.weight", "conv3.0.downsample.1.weight", "conv3.0.downsample.1.bias", "conv3.0.downsample.1.running_mean", "conv3.0.downsample.1.running_var", "conv3.1.conv1.weight", "conv3.1.bn1.weight", "conv3.1.bn1.bias", "conv3.1.bn1.running_mean", "conv3.1.bn1.running_var", "conv3.1.conv2.weight", "conv3.1.bn2.weight", "conv3.1.bn2.bias", "conv3.1.bn2.running_mean", "conv3.1.bn2.running_var", "conv3.1.conv3.weight", "conv3.1.bn3.weight", "conv3.1.bn3.bias", "conv3.1.bn3.running_mean", "conv3.1.bn3.running_var", "conv3.1.se_module.fc1.weight", "conv3.1.se_module.fc1.bias", "conv3.1.se_module.fc2.weight", "conv3.1.se_module.fc2.bias", "conv3.2.conv1.weight", "conv3.2.bn1.weight", "conv3.2.bn1.bias", "conv3.2.bn1.running_mean", "conv3.2.bn1.running_var", "conv3.2.conv2.weight", "conv3.2.bn2.weight", "conv3.2.bn2.bias", "conv3.2.bn2.running_mean", "conv3.2.bn2.running_var", "conv3.2.conv3.weight", "conv3.2.bn3.weight", "conv3.2.bn3.bias", "conv3.2.bn3.running_mean", "conv3.2.bn3.running_var", "conv3.2.se_module.fc1.weight", "conv3.2.se_module.fc1.bias", "conv3.2.se_module.fc2.weight", "conv3.2.se_module.fc2.bias", "conv3.3.conv1.weight", "conv3.3.bn1.weight", "conv3.3.bn1.bias", "conv3.3.bn1.running_mean", "conv3.3.bn1.running_var", "conv3.3.conv2.weight", "conv3.3.bn2.weight", "conv3.3.bn2.bias", "conv3.3.bn2.running_mean", "conv3.3.bn2.running_var", "conv3.3.conv3.weight", "conv3.3.bn3.weight", "conv3.3.bn3.bias", "conv3.3.bn3.running_mean", "conv3.3.bn3.running_var", "conv3.3.se_module.fc1.weight", "conv3.3.se_module.fc1.bias", "conv3.3.se_module.fc2.weight", "conv3.3.se_module.fc2.bias", "conv4.0.conv1.weight", "conv4.0.bn1.weight", "conv4.0.bn1.bias", "conv4.0.bn1.running_mean", "conv4.0.bn1.running_var", "conv4.0.conv2.weight", "conv4.0.bn2.weight", "conv4.0.bn2.bias", "conv4.0.bn2.running_mean", "conv4.0.bn2.running_var", "conv4.0.conv3.weight", "conv4.0.bn3.weight", "conv4.0.bn3.bias", "conv4.0.bn3.running_mean", "conv4.0.bn3.running_var", "conv4.0.se_module.fc1.weight", "conv4.0.se_module.fc1.bias", "conv4.0.se_module.fc2.weight", "conv4.0.se_module.fc2.bias", "conv4.0.downsample.0.weight", "conv4.0.downsample.1.weight", "conv4.0.downsample.1.bias", "conv4.0.downsample.1.running_mean", "conv4.0.downsample.1.running_var", "conv4.1.conv1.weight", "conv4.1.bn1.weight", "conv4.1.bn1.bias", "conv4.1.bn1.running_mean", "conv4.1.bn1.running_var", "conv4.1.conv2.weight", "conv4.1.bn2.weight", "conv4.1.bn2.bias", "conv4.1.bn2.running_mean", "conv4.1.bn2.running_var", "conv4.1.conv3.weight", "conv4.1.bn3.weight", "conv4.1.bn3.bias", "conv4.1.bn3.running_mean", "conv4.1.bn3.running_var", "conv4.1.se_module.fc1.weight", "conv4.1.se_module.fc1.bias", "conv4.1.se_module.fc2.weight", "conv4.1.se_module.fc2.bias", "conv4.2.conv1.weight", "conv4.2.bn1.weight", "conv4.2.bn1.bias", "conv4.2.bn1.running_mean", "conv4.2.bn1.running_var", "conv4.2.conv2.weight", "conv4.2.bn2.weight", "conv4.2.bn2.bias", "conv4.2.bn2.running_mean", "conv4.2.bn2.running_var", "conv4.2.conv3.weight", "conv4.2.bn3.weight", "conv4.2.bn3.bias", "conv4.2.bn3.running_mean", "conv4.2.bn3.running_var", "conv4.2.se_module.fc1.weight", "conv4.2.se_module.fc1.bias", "conv4.2.se_module.fc2.weight", "conv4.2.se_module.fc2.bias", "conv4.3.conv1.weight", "conv4.3.bn1.weight", "conv4.3.bn1.bias", "conv4.3.bn1.running_mean", "conv4.3.bn1.running_var", "conv4.3.conv2.weight", "conv4.3.bn2.weight", "conv4.3.bn2.bias", "conv4.3.bn2.running_mean", "conv4.3.bn2.running_var", "conv4.3.conv3.weight", "conv4.3.bn3.weight", "conv4.3.bn3.bias", "conv4.3.bn3.running_mean", "conv4.3.bn3.running_var", "conv4.3.se_module.fc1.weight", "conv4.3.se_module.fc1.bias", "conv4.3.se_module.fc2.weight", "conv4.3.se_module.fc2.bias", "conv5.0.conv1.weight", "conv5.0.bn1.weight", "conv5.0.bn1.bias", "conv5.0.bn1.running_mean", "conv5.0.bn1.running_var", "conv5.0.conv2.weight", "conv5.0.bn2.weight", "conv5.0.bn2.bias", "conv5.0.bn2.running_mean", "conv5.0.bn2.running_var", "conv5.0.conv3.weight", "conv5.0.bn3.weight", "conv5.0.bn3.bias", "conv5.0.bn3.running_mean", "conv5.0.bn3.running_var", "conv5.0.se_module.fc1.weight", "conv5.0.se_module.fc1.bias", "conv5.0.se_module.fc2.weight", "conv5.0.se_module.fc2.bias", "conv5.0.downsample.0.weight", "conv5.0.downsample.1.weight", "conv5.0.downsample.1.bias", "conv5.0.downsample.1.running_mean", "conv5.0.downsample.1.running_var", "conv5.1.conv1.weight", "conv5.1.bn1.weight", "conv5.1.bn1.bias", "conv5.1.bn1.running_mean", "conv5.1.bn1.running_var", "conv5.1.conv2.weight", "conv5.1.bn2.weight", "conv5.1.bn2.bias", "conv5.1.bn2.running_mean", "conv5.1.bn2.running_var", "conv5.1.conv3.weight", "conv5.1.bn3.weight", "conv5.1.bn3.bias", "conv5.1.bn3.running_mean", "conv5.1.bn3.running_var", "conv5.1.se_module.fc1.weight", "conv5.1.se_module.fc1.bias", "conv5.1.se_module.fc2.weight", "conv5.1.se_module.fc2.bias", "conv5.2.conv1.weight", "conv5.2.bn1.weight", "conv5.2.bn1.bias", "conv5.2.bn1.running_mean", "conv5.2.bn1.running_var", "conv5.2.conv2.weight", "conv5.2.bn2.weight", "conv5.2.bn2.bias", "conv5.2.bn2.running_mean", "conv5.2.bn2.running_var", "conv5.2.conv3.weight", "conv5.2.bn3.weight", "conv5.2.bn3.bias", "conv5.2.bn3.running_mean", "conv5.2.bn3.running_var", "conv5.2.se_module.fc1.weight", "conv5.2.se_module.fc1.bias", "conv5.2.se_module.fc2.weight", "conv5.2.se_module.fc2.bias", "fc.0.weight", "fc.0.bias", "id_fc.0.weight", "id_fc.0.bias". 
	Unexpected key(s) in state_dict: "module.first_bn.weight", "module.first_bn.bias", "module.first_bn.running_mean", "module.first_bn.running_var", "module.first_bn.num_batches_tracked", "module.encoder.layer0.conv1.weight", "module.encoder.layer0.bn1.weight", "module.encoder.layer0.bn1.bias", "module.encoder.layer0.bn1.running_mean", "module.encoder.layer0.bn1.running_var", "module.encoder.layer0.bn1.num_batches_tracked", "module.encoder.layer1.0.conv1.weight", "module.encoder.layer1.0.bn1.weight", "module.encoder.layer1.0.bn1.bias", "module.encoder.layer1.0.bn1.running_mean", "module.encoder.layer1.0.bn1.running_var", "module.encoder.layer1.0.bn1.num_batches_tracked", "module.encoder.layer1.0.conv2.weight", "module.encoder.layer1.0.bn2.weight", "module.encoder.layer1.0.bn2.bias", "module.encoder.layer1.0.bn2.running_mean", "module.encoder.layer1.0.bn2.running_var", "module.encoder.layer1.0.bn2.num_batches_tracked", "module.encoder.layer1.0.conv3.weight", "module.encoder.layer1.0.bn3.weight", "module.encoder.layer1.0.bn3.bias", "module.encoder.layer1.0.bn3.running_mean", "module.encoder.layer1.0.bn3.running_var", "module.encoder.layer1.0.bn3.num_batches_tracked", "module.encoder.layer1.0.se_module.fc1.weight", "module.encoder.layer1.0.se_module.fc1.bias", "module.encoder.layer1.0.se_module.fc2.weight", "module.encoder.layer1.0.se_module.fc2.bias", "module.encoder.layer1.0.downsample.0.weight", "module.encoder.layer1.0.downsample.1.weight", "module.encoder.layer1.0.downsample.1.bias", "module.encoder.layer1.0.downsample.1.running_mean", "module.encoder.layer1.0.downsample.1.running_var", "module.encoder.layer1.0.downsample.1.num_batches_tracked", "module.encoder.layer1.1.conv1.weight", "module.encoder.layer1.1.bn1.weight", "module.encoder.layer1.1.bn1.bias", "module.encoder.layer1.1.bn1.running_mean", "module.encoder.layer1.1.bn1.running_var", "module.encoder.layer1.1.bn1.num_batches_tracked", "module.encoder.layer1.1.conv2.weight", "module.encoder.layer1.1.bn2.weight", "module.encoder.layer1.1.bn2.bias", "module.encoder.layer1.1.bn2.running_mean", "module.encoder.layer1.1.bn2.running_var", "module.encoder.layer1.1.bn2.num_batches_tracked", "module.encoder.layer1.1.conv3.weight", "module.encoder.layer1.1.bn3.weight", "module.encoder.layer1.1.bn3.bias", "module.encoder.layer1.1.bn3.running_mean", "module.encoder.layer1.1.bn3.running_var", "module.encoder.layer1.1.bn3.num_batches_tracked", "module.encoder.layer1.1.se_module.fc1.weight", "module.encoder.layer1.1.se_module.fc1.bias", "module.encoder.layer1.1.se_module.fc2.weight", "module.encoder.layer1.1.se_module.fc2.bias", "module.encoder.layer1.2.conv1.weight", "module.encoder.layer1.2.bn1.weight", "module.encoder.layer1.2.bn1.bias", "module.encoder.layer1.2.bn1.running_mean", "module.encoder.layer1.2.bn1.running_var", "module.encoder.layer1.2.bn1.num_batches_tracked", 

报错原因是加载过程中,期望获得的key值为feature…,而不是module.features…。这是由模型保存过程中导致的,模型应该是在DataParallel模式下面,也就是采用了多GPU训练模型,然后直接保存的。
详细参考博客:https://blog.csdn.net/yangwangnndd/article/details/100207686
解决方法将key值中的module去掉或者加入并行都可以,

initial_checkpoint = config.pretrained_model

network = Net(num_class=2, is_first_bn=True)
    # network = CDCNpp().to(device)
network = torch.nn.DataParallel(network)
network = network.cuda()
network.eval()
state = torch.load(model_name)
# network.load_state_dict({k.replace('module.', ''): v for k, v in state['state_dict'].items()})
network.load_state_dict(torch.load(initial_checkpoint, map_location=lambda storage, loc: storage))
# network.load_state_dict({k.replace('module.', ''): v for k, v in torch.load(initial_checkpoint).items()})
  • 4
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值