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

相关推荐
蚂蚁在飞-3 分钟前
浏览器自动化操作AI工具-browser-use
运维·人工智能·自动化
jndingxin18 分钟前
OpenCV 图形API(17)计算输入矩阵 src 中每个元素的平方根函数sqrt()
人工智能·opencv
Lilith的AI学习日记22 分钟前
LangChain核心架构解析:从传统Chain到LCEL的演进之路
人工智能·架构·langchain·ai编程
qq_2739002327 分钟前
Pytorch torch.nn.utils.rnn.pad_sequence 介绍
人工智能·pytorch·python·rnn·深度学习
气π31 分钟前
【Vue-组件】学习笔记
vue.js·笔记·学习
大模型真好玩35 分钟前
点名DeepSeek?一文解析”最强大模型“Llama-4到底是王者归来 还是 困兽犹斗?
人工智能·llama·deepseek
shengjk139 分钟前
快速搞懂湖仓一体发展历程:错过它,你将失去下一个大数据风口!
人工智能·后端
拉丁解牛说技术1 小时前
一句话让deepseek实现基于Spring AI的AI chatbot
人工智能·spring·程序员
在下小航1 小时前
前端本地大模型 window.ai 最新教程
前端·人工智能
蹦蹦跳跳真可爱5891 小时前
Python----PaddlePaddle(深度学习框架PaddlePaddle,概述,安装,衍生工具)
开发语言·人工智能·python·paddlepaddle