问题分析
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去掉即可
问题分析
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去掉即可