深度学习加载GPU并行训练后的模型model.load_state_dict(torch.load(path)) 导致的问题

问题描述
1、Missing key(s) in state_dict
2、测试结果全部为0或者全部为某一固定值

解决方法：
checkpoint = torch.load(path)
model.load_state_dict({k.replace(‘module.’, ‘’): v for k, v in checkpoint.items()}, strict=False)