训练模型转换
Linux
环境准备
模型导出
模型转换
中级
高级
概述
创建MindSpore端侧模型的步骤:
首先基于MindSpore架构使用Python创建网络模型,并导出为
.mindir
文件,参见云端的保存模型。然后将
.mindir
模型文件转换成.ms
文件,.ms
文件可以导入端侧设备并基于MindSpore端侧框架训练。
Linux环境
环境准备
MindSpore Lite 模型转换工具提供了多个参数,目前工具仅支持Linux系统,环境准备步骤:
编译或下载已编译好的模型转换工具。
配置模型转换工具的环境变量。
参数说明
下表为MindSpore Lite训练模型转换工具使用到的参数:
参数 |
是否必选 |
参数说明 |
取值范围 |
默认值 |
---|---|---|---|---|
|
否 |
打印全部帮助信息 |
- |
- |
|
是 |
输入模型的原始格式 |
MINDIR |
- |
|
是 |
MINDIR模型文件名(包括路径) |
- |
- |
|
是 |
输出模型文件名(包括路径)自动生成 |
- |
- |
|
是 |
是否是训练模式;如果要训练模型,必须为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:
程序会返回的错误码和错误信息。