昇思25天学习打卡营第七天|保存与加载

       今天学习如何第9课，保存与加载模型的相关知识。

 

 

      首先是保存与加载模型权重，在保存模型的时候使用save_checkpoint接口，传入网络和指定的保存路径。先创造相同模型的实例，然后再使用load_checkpoint和load_param_into_net方法加载参数。

      在保存与加载MindIR中，可使用export接口直接将模型保存为MindIR。它同时保存了checkpoint和模型结构，因此需要定义输入Tensor来获取输入shape。