前言

模型的數據需要存儲和加載，這節介紹存儲和加載的方式方法。

存和加載模型權重

保存模型使用save_checkpoint接口，傳入網絡和指定的保存路徑，要加載模型權重，需要先創建相同模型的實例，然后使用load_checkpoint和load_param_into_net方法加載參數。

保存和加載文件

保存和加載MindIR

除Checkpoint外，MindSpore提供了云側（訓練）和端側（推理）統一的中間表示（Intermediate Representation，IR）。可使用export接口直接將模型保存為MindIR。

已有的MindIR模型可以方便地通過load接口加載，傳入nn.GraphCell即可進行推理。

MindIR同時保存了Checkpoint和模型結構，因此需要定義輸入Tensor來獲取輸入shape。

將模型保存為MindIR。

總結

這篇介紹了，在MindSpore中保存和加載模型的方法,包括使用save_checkpoint保存模型參數,load_checkpoint和load_param_into_net加載參數,以及使用export導出MindIR格式模型并通過load接口加載進行推理。還介紹了這些保存和加載模型的核心API,為開發者提供了在MindSpore中管理和復用模型的實用技巧。