昇思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装饰器,配置为静态图模式

相关推荐
Ray Song1 分钟前
自动驾驶传感器数据录制过程中的五大系统性挑战
人工智能·自动驾驶·数据录制
weixin_549808366 分钟前
从效能革命到体验重构,易路 AI Agent 破局 HR 三重困境
大数据·人工智能·重构
量子位8 分钟前
刚得诺奖的成果被做成芯片了
人工智能·ai编程
新智元30 分钟前
复旦张军平:人类是硅基生命过渡体?
人工智能·openai
东风西巷35 分钟前
MobiPDF安卓版(PDF阅读编辑工具) 修改版
学习·pdf·电脑·软件需求
新智元36 分钟前
谷歌Gemini 3.0「全家桶」年度压轴,前端不再需要人类!下周王者降临
人工智能·openai
CVer_1 小时前
Pytorch学习之Dataset类
学习
机器之心1 小时前
ICLR 2026惊现SAM 3,分割一切的下一步:让模型理解「概念」
人工智能·openai
清风吹过1 小时前
LSTM新架构论文分享6:LSTM+Transformer融合
论文阅读·人工智能·深度学习·神经网络·lstm·transformer
Hoking1 小时前
LangChain4j集成SpringBoot接入百炼大模型(Qwen)
java·人工智能·spring boot·llm