Pytorch 调试错误记录

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

>>阅读原文<<