发现问题
载入模型时发现模型的参数不匹配:
探究问题
打印出权重文件里面每一层的参数:
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)
问题解决