mirror of https://github.com/alibaba/MNN.git
1.2 KiB
1.2 KiB
MNN 训练相关工具
编译
MNN 编译与安装
- MNN 编译时打开 MNN_BUILD_TRAIN 开关：cmake .. -DMNN_BUILD_TRAIN=true
产物
- transformer
- extractForInfer
- runTrainDemo.out
使用
制作训练模型
eg: ./transformer mobilenet.mnn mobilenet-train.mnn transformerConfig.json [revert.json]
- 第一个参数：输入，推理模型或计算Loss的模型
- 第二个参数：输出，训练模型
- 第三个参数：输入，为json配置文件, 参考 transformerConfig.json 和 transformerConfig2.json 编写
- 第四个参数：输出，还原参数所需要的配置文件，若不指定，默认为当前路径下的 revert.json
制作训练数据
- 根据模型输入和标注准备
训练
- 用 Interpreter-Session API 加载训练模型，得到 Session
- 根据 transformer 打印的输入，传入 Session
- 运行 Session
- （可选）获取 Session 中的 loss 并查看
- 若干次迭代后，调用 Interpreter 的 updateSessionToModel 更新权重, 然后调用 getModelBuffer 获取新的模型内存并自行写入到新的模型文件中
应用
- 使用 extractForInfer 将训练模型的参数提取到 推理模型
- 重新加载推理模型，即训练好的模型