学习目标
day7学习了如何调整超参数,并进行网络模型训练。
day8需要学习如何保存与加载模型,以用于微调和后续的模型推理与部署。
模型权重
保存模型使用save_checkpoint
接口,传入网络和指定的保存路径。
加载模型权重,需要先创建相同模型的实例,再使用load_checkpoint
和load_param_into_net
方法加载参数。
MindIR
MindSpore提供了云侧(训练)和端侧(推理)统一的中间表达,因此使用export
接口可以直接将模型保存为MindIR。