Pytorch 调试错误记录

Pytorch 调试错误记录

1. 加载模型错误，加载模型参数时，关键字不匹配，每个都多了“module”
   
   解决方法：给自己的net再套一层torch.nn.DataParallel()就行，或者在你训练的时候把这句话去掉，这句话是在你有多卡情况下用的，如果你只有一张卡可以直接去掉。