Hello ptrblck,
I am exactly following same process for model loading
model=CQCCModel()
optimizer = torch.optim.Adam(model.parameters(), lr=0.0001)
model.load_state_dict(torch.load(model_path, map_location=‘cuda’))
optimizer.load_state_dict(torch.load(model_path, map_location=‘cuda’))
but I am getting this errors:
self.class.name, “\n\t”.join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for CQCCModel:
Missing key(s) in state_dict: “layer1.0.weight”, “layer1.0.bias”, “layer1.1.weight”, “layer1.1.bias”, “layer1.1.running_mean”, “layer1.1.running_var”, “layer2.0.conv1.weight”, “layer2.0.conv1.bias”, “layer2.0.bn1.weight”, “layer2.0.bn1.bias”, “layer2.0.bn1.running_mean”, “layer2.0.bn1.running_var”, “layer2.0.conv2.weight”, “layer2.0.conv2.bias”, “layer2.0.conv11.weight”, “layer2.0.conv11.bias”, “layer3.0.conv1.weight”, “layer3.0.conv1.bias”, “layer3.0.bn1.weight”, “layer3.0.bn1.bias”, “layer3.0.bn1.running_mean”, “layer3.0.bn1.running_var”, “layer3.0.conv2.weight”, “layer3.0.conv2.bias”, “layer3.0.conv11.weight”, “layer3.0.conv11.bias”, “layer3.0.pre_bn.weight”, “layer3.0.pre_bn.bias”, “layer3.0.pre_bn.running_mean”, “layer3.0.pre_bn.running_var”, “layer4.0.conv1.weight”,
Unexpected key(s) in state_dict: "module.layer1.0.weight", "module.layer1.0.bias", "module.layer1.1.weight", "module.layer1.1.bias", "module.layer1.1.running_mean", "module.layer1.1.running_var", "module.layer1.1.num_batches_tracked", "module.layer2.0.conv1.weight", "module.layer2.0.conv1.bias", "module.layer2.0.bn1.weight", "module.layer2.0.bn1.bias", "module.layer2.0.bn1.running_mean", "module.layer2.0.bn1.running_var", "module.layer2.0.bn1.num_batches_tracked", "module.layer2.0.conv2.weight", "module.layer2.0.conv2.bias", "module.layer2.0.conv11.weight", "module.layer2.0.conv11.bias", "module.layer3.0.conv1.weight", "module.layer3.0.conv1.bias", "module.layer3.0.bn1.weight", "module.layer3.0.bn1.bias", "module.layer3.0.bn1.running_mean", "module.layer3.0.bn1.running_var", "module.layer3.0.bn1.num_batches_tracked", "module.layer3.0.conv2.weight", "module.layer3.0.conv2.bias", "module.layer3.0.conv11.weight", "module.layer3.0.conv11.bias", "module.layer3.0.pre_bn.weight", "module.layer3.0.pre_bn.bias", "module.layer3.0.pre_bn.running_mean", "module.layer3.0.pre_bn.running_var", "module.layer3.0.pre_bn.num_batches_tracked", "module.layer4.0.conv1.weight", "module.layer4.0.conv1.bias", "module.layer4.0.bn1.weight", "module.layer4.0.bn1.bias", "module.layer4.0.bn1.running_mean", "module.layer4.0.bn1.running_var", "module.layer4.0.bn1.num_batches_tracked", "module.layer4.0.conv2.weight", "module.layer4.0.conv2.bias", "module.layer4.0.conv11.weight", "module.layer4.0.conv11.bias"
I save the model in this way:
model = CQCCModel()
torch.save(model.state_dict(), os.path.join(model_save_path, ‘epoch_{}.pth’.format(epoch)))
I didn’t understand this error.
Could you please let me know why this kind of error is coming and what is the right way to load the model.
Thanks in advance.