昇思MindSpore打卡营 11号 -保存与加载 学习心得

目的

保存训练后的结果，以便后续加载用于微调和后续的模型推理与部署。

关于模型权重

使用save_checkpoint接口，传入网络和指定的保存路径进行保存；

创建相同模型的实例，然后使用load_checkpoint和load_param_into_net方法加载参数，来进行加载模型权重。

关于MindIR

MindIR同时保存了Checkpoint和模型结构，可以使用export接口直接将模型保存为MindIR。

可以通过load接口加载MindIR模型，进行推理时，传入nn.GraphCell即可。