深度学习中加载权重文件时，权重参数前面出现module.怎么办

发现问题

载入模型时发现模型的参数不匹配：

探究问题

打印出权重文件里面每一层的参数：

checkpoint  = torch.load('model_2440_0.0193.pth.tar', map_location = 'cpu')
for key in checkpoint:
    print(key)

得到：

发现每一个权重文件前面都有module.
原因：dist模式训练的模型很可能module

解决办法

new_state_dict = {}
for k,v in checkpoint.items():
    new_state_dict[k[7:]] = v
net.load_state_dict(new_state_dict)

问题解决