今天进入了初学入门初学教程的保存与加载的学习,下面我们来看看吧。
保存与加载
在训练完模型之后,我们主要是保存结果,用于微调和后续的模型推理与部署,因此很有必要学习如何正确的保存和加载,助推AI的调试。
如图所说,保存模型使用save_checkpoint
接口,传入网络和指定的保存路径,而要加载模型权重,需要先创建相同模型的实例,然后使用load_checkpoint
和load_param_into_net
方法加载参数。
其次要保存和加载MindIR,因为MindSpore提供了云侧(训练)和端侧(推理)统一的中间表示(Intermediate Representation,IR)所以可使用export
接口直接将模型保存为MindIR。
最后附上打卡时间
学习打卡结束