第八天
本节学习了保存与加载模型。
1.保存和加载模型权重
保存模型使用save_checkpoint接口,传入网络和指定的保存路径,要加载模型权重,需要先创建相同模型的实例,然后使用load_checkpoint和load_param_into_net方法加载参数
2.保存和加载MindIR
除Checkpoint外,MindSpore提供了云侧(训练)和端侧(推理)统一的中间表示。可使用export接口直接将模型保存为MindIR。
《昇思25天学习打卡营第8天 | 昇思MindSpore保存与加载》
最新推荐文章于 2024-07-25 22:16:54 发布