在 pytorch 多GPU训练下，存储 整个模型 （ 而不是model.state_dict() ）后再调用模型可能会遇到下面的情况：

AttributeError: ‘DataParallel’ object has no attribute ‘xxxx’

解决的方法是：

model = torch.load('path/to/model')
#添加语句部分
if isinstance(model,torch.nn.DataParallel):
		model = model.module

#下面就可以正常使用了
model.eval()
...
...
...
Logo
2048 AI社区

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

更多推荐

苍穹外卖day11

*解答**：这是一个典型的**场景化数据展示**问题。**解答**：这是一个**空对象更新**的风险点！// 没看到需要订单菜品信息啊。**教训**：前端传参可能只包含必要字段（这里只有订单ID），后端应根据**业务语义**确定状态值，而不是盲目依赖前端传入的状态。**分析**：待付款订单的支付状态为`UN_PAID`，不满足`PAID`条件，因此无法进入取消逻辑。**问题**：为什么`order

avatar 2048 AI社区

使用Bolt.new快速开发一个博客网站

Bolt.new是一个基于AI的全栈Web开发平台，用户只需用自然语言描述需求即可自动生成完整网站代码，支持主流框架并可直接部署。开发流程包括：1)登录创建项目；2)输入需求描述生成初始代码；3)通过对话优化样式和内容；4)实时预览迭代；5)一键部署上线。该平台特别适合无代码用户快速搭建博客、商城等原型，复杂功能可能需要多轮调整或手动编辑。所有开发都在浏览器中完成，无需本地环境配置。

avatar 2048 AI社区
cover

Python Web 开发进阶实战：数字孪生平台 —— 在 Flask + Vue 中构建实时物理世界镜像

avatar 2048 AI社区