训练模型转换

Linux 环境准备 模型导出 模型转换 中级 高级

• 训练模型转换

  • 概述

  • Linux环境

    • 环境准备

    • 参数说明

    • 模型转换示例

概述

创建MindSpore端侧模型的步骤：

• 首先基于MindSpore架构使用Python创建网络模型，并导出为.mindir文件，参见云端的保存模型。

• 然后将.mindir模型文件转换成.ms文件，.ms文件可以导入端侧设备并基于MindSpore端侧框架训练。

Linux环境

环境准备

MindSpore Lite 模型转换工具提供了多个参数，目前工具仅支持Linux系统，环境准备步骤：

• 编译或下载已编译好的模型转换工具。

• 配置模型转换工具的环境变量。

参数说明

下表为MindSpore Lite训练模型转换工具使用到的参数：

参数	是否必选	参数说明	取值范围	默认值
--help	否	打印全部帮助信息	-	-
--fmk=<FMK>	是	输入模型的原始格式	MINDIR	-
--modelFile=<MODELFILE>	是	MINDIR模型文件名（包括路径）	-	-
--outputFile=<OUTPUTFILE>	是	输出模型文件名（包括路径）自动生成.ms后缀	-	-
--trainModel=true	是	是否是训练模式；如果要训练模型，必须为true	true, false	false

参数名称和数值之间使用等号连接且不能有空格。

模型转换示例

假设待转换的模型文件为my_model.mindir，执行如下转换命令：

./converter_lite --fmk=MINDIR --trainModel=true --modelFile=my_model.mindir --outputFile=my_model

转换成功输出如下：

CONVERTER RESULT SUCCESS:0

这表明 MindSpore 模型成功转换为 MindSpore 端侧模型，并生成了新文件my_model.ms。如果转换失败输出如下：

CONVERT RESULT FAILED:

程序会返回的错误码和错误信息。