昇思25天学习打卡营第8天|保存与加载

当模型训练好以后,就可以用来预测了。后续我们在其他端部署的时候不需要再次训练,直接使用训练好的模型进行推理就可以了。

在保存和加载的时候我们都需要传入模型。保存的API是save_checkpoint(model,path), 对应的加载的API是 load_checkpoint(model,path)

除Checkpoint外,MindSpore提供了云侧(训练)和端侧(推理)统一的中间表示(Intermediate Representation,IR)。可使用export接口直接将模型保存为MindIR

使用静态图加速

前面提到过,mindspore有两种运行模式。动态图和静态图。

动态图类似于debug模式,静态图类似于release模式。

可以使用如下代码配置为静态图模式,来获得更快的训练速度。

python 复制代码
import mindspore as ms
ms.set_context(mode=ms.GRAPH_MODE) 

也可以使用@ms.hit装饰器,配置为静态图模式

相关推荐
JVM跑得快18 分钟前
Python 基础语法 × Java 对比
人工智能
AI攻城狮24 分钟前
Kimi Bot + OpenClaw 完整配置指南:5 步实现本地 AI Agent 集成
人工智能·云原生·aigc
货拉拉技术1 小时前
货拉拉海豚平台-大模型推理加速工程化实践
人工智能·后端·架构
掘金安东尼1 小时前
国内大模型真实格局:用户规模 vs API调用量(v2026.3.6)
人工智能
前端一课2 小时前
OpenClaw 项目全面架构分析报告
前端·人工智能
小姐姐味道2 小时前
1万美金的账单,烧麻了!bull这个skills让数据推理质量更好,更省!
人工智能
苍何2 小时前
再见 Openclaw,桌面端 Agent 起飞了!
人工智能
雮尘2 小时前
让 AI Agent 高效并行开发的命令-git worktree
人工智能·git·agent
Ray Liang3 小时前
AI基于Spec开发是巨坑?
人工智能·架构设计·mindx
哔哩哔哩技术3 小时前
游戏数据分析Agent的全栈架构演进
人工智能·agent