使用torch.nn.DataParallel进行多核训练或测试时,调用网络模块报错
AttributeError: ‘DataParallel’ object has no attribute ‘xxxx’
解决措施:
在调用网络模块出如model.XXX换成model.module.XXX。
AttributeError: ‘DataParallel‘ object has no attribute ‘xxxx‘
最新推荐文章于 2024-09-11 20:13:23 发布
在使用PyTorch的DataParallel进行多核训练或测试时,如果遇到AttributeError,比如DataParallelobjecthasnoattributexxx,解决方案是将对模型模块的调用从model.XXX改为model.module.XXX。这样做是因为DataParallel会包装原始模型,而module属性可以访问到原始未包装的模型。
摘要由CSDN通过智能技术生成