多卡训练转换到单卡训练，在保存模型时发生了错误：

torch.nn.modules.module.ModuleAttributeError: 'VGG' object has no attribute 'module'

后来发现是保存模型的时候用了torch.module.statedict()

stateG = {'model': netG.module.cpu().state_dict(), 'optimizer': optimizerG.state_dict(), 'epoch': epoch}
torch.save(stateG, net_g_model_out_path)

上网查找资料后将代码改为：

stateG = {'model': netG.cpu().state_dict(), 'optimizer': optimizerG.state_dict(), 'epoch': epoch}
torch.save(stateG, net_g_model_out_path)

# pytorch # 深度学习 # python
Logo
2048 AI社区

有“AI”的1024 = 2048，欢迎大家加入2048 AI社区

更多推荐

Chaterm致力于打造20年经验的SRE副驾驶

合合信息推出的Chaterm定位为"20年经验的运维专家"，旨在解决AI时代复杂的运维挑战。该产品能通过模糊问题描述快速定位服务器故障，利用多维度并行分析能力大幅缩短故障解决时间（MTTR）。在云原生环境中，Chaterm作为智能自动化层，帮助管理异构基础设施（GPU/NPU/CUDA等），应对微服务和K8s体系下的立体化故障排查难题。其开箱即用的特性降低了企业交付成本，已在故

avatar 2048 AI社区
cover

Java 部署：应用性能监控（New Relic/APM 配置）

avatar 2048 AI社区

初探AI世界

人工智能（AI）是通过程序让机器模拟人类思维和行为的计算机技术，可分为弱AI、强AI和超AI三类。大模型是AI的高级形态，经过预训练和微调等过程，具备知识处理和推理能力。AI生态包含大模型开发者、应用开发者和使用者三个角色。AI Agent是具备自主决策和执行能力的智能实体，由大模型、记忆、任务规划和工具使用等组件构成，能通过多Agent协作完成复杂任务。AI技术正深刻影响电商、教育、金融等行业，

avatar 2048 AI社区