问题分析
fd.write(str(model.module.netG))
fd.write(str(model.module.netD)) 是多GPU模型代码,自己的设备是单GPU,不能使用此句
改成
fd.write(str(model.netG))
fd.write(str(model.netD))
问题解决
单GPU电脑跑模型,将.module去掉即可
文章讲述了在使用多GPU模型的代码中,对于单GPU设备用户遇到的问题,即如何将`model.module.netG`和`model.module.netD`修改为`model.netG`和`model.netD`来适应单GPU环境。解决方案只需移除`.module`即可运行。
问题分析
fd.write(str(model.module.netG))
fd.write(str(model.module.netD)) 是多GPU模型代码,自己的设备是单GPU,不能使用此句
改成
fd.write(str(model.netG))
fd.write(str(model.netD))
问题解决
单GPU电脑跑模型,将.module去掉即可
1189

被折叠的 条评论
为什么被折叠?