Missing key(s) in state_dict与 Unexpected key(s) in state_dict

Missing key(s) in state_dict: "model.0.0.weight", "model.0.1.weight", ……

Unexpected key(s) in state_dict: "module.model.0.0.weight", "module.model.0.1.weight",……

保存模型时用了多gpu：model = nn.DataParallel(model, device_ids=[6,7])

在读取时也加上这句即可。