您的位置: 首页 > 文章 > Pytorch 调试错误记录

Pytorch 调试错误记录

分类: 文章 • 2025-02-18 17:58:52

加载模型错误，加载模型参数时，关键字不匹配，每个都多了“module”

解决方法：给自己的net再套一层torch.nn.DataParallel()就行，或者在你训练的时候把这句话去掉，这句话是在你有多卡情况下用的，如果你只有一张卡可以直接去掉。