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

相关推荐
KG_LLM图谱增强大模型34 分钟前
Vgent:基于图的多模态检索推理增强生成框架GraphRAG,突破长视频理解瓶颈
大数据·人工智能·算法·大模型·知识图谱·多模态
AKAMAI35 分钟前
企业如何平衡AI创新与风险
人工智能·云原生·云计算
AA陈超37 分钟前
UE5笔记:GetWorld()->SpawnActorDeferred()
c++·笔记·学习·ue5·虚幻引擎
生椰拿铁You2 小时前
openxlpy学习笔记
笔记·学习
TDengine (老段)2 小时前
优化 TDengine IDMP 面板编辑的几种方法
人工智能·物联网·ai·时序数据库·tdengine·涛思数据
AA陈超2 小时前
ASC学习笔记0025:移除所有属性集
c++·笔记·学习·ue5·虚幻引擎
数据的世界013 小时前
Visual Studio 2026 正式发布:AI 原生 IDE 与性能革命的双向突破
ide·人工智能·visual studio
shayudiandian4 小时前
深度学习中的激活函数全解析:该选哪一个?
人工智能·深度学习
视界先声4 小时前
如何选择合适的养老服务机器人
人工智能·物联网·机器人