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

相关推荐
测试人社区-千羽4 分钟前
生物识别系统的测试安全性与漏洞防护实践
运维·人工智能·opencv·安全·数据挖掘·自动化·边缘计算
d111111111d6 分钟前
STM32得中断服务函数,为什么不能有返回值
笔记·stm32·单片机·嵌入式硬件·学习
2501_924794906 分钟前
企业AI转型为何难?——从“不敢用”到“用得稳”的路径重构
大数据·人工智能·重构
阿蒙Amon15 分钟前
JavaScript学习笔记:12.类
javascript·笔记·学习
Tezign_space18 分钟前
小红书内容运营工具怎么选?专业视角拆解优质工具核心标准
大数据·人工智能·内容运营
光影少年19 分钟前
PostgreSQL数据库学习路线
数据库·学习·postgresql
老马啸西风20 分钟前
成熟企业级技术平台 MVE-010-跳板机 / 堡垒机(Jump Server / Bastion Host)
人工智能·深度学习·算法·职场和发展
康实训20 分钟前
养老实训室建设标准指南
大数据·人工智能·实训室·养老实训室·实训室建设
袖手蹲21 分钟前
Arduino UNO Q 烘托圣诞节气氛
人工智能·单片机·嵌入式硬件
wjykp28 分钟前
part 3神经网络的学习
人工智能·神经网络·学习