使用MindSpore Lite进行端侧训练 (C/C++)
==============================

#### 场景介绍

MindSpore Lite是一款AI引擎，它提供了面向不同硬件设备AI模型推理的功能，目前已经在图像分类、目标识别、人脸识别、文字识别等应用中广泛使用，同时支持在端侧设备上进行部署训练，让模型在实际业务场景中自适应用户的行为。

本文介绍使用MindSpore Lite端侧AI引擎进行模型训练的通用开发流程。

#### 接口说明

此处给出使用MindSpore Lite进行模型训练相关的部分接口，具体请见下方表格。更多接口及详细内容，请见[MindSpore](https://developer.huawei.com/consumer/cn/doc/harmonyos-references/capi-mindspore)。

| 接口名称 | 描述 |
| :-- | :-- |
| OH\_AI\_ContextHandle OH\_AI\_ContextCreate() | 创建一个上下文的对象。注意：此接口需跟OH\_AI\_ContextDestroy配套使用。 |
| OH\_AI\_DeviceInfoHandle OH\_AI\_DeviceInfoCreate(OH\_AI\_DeviceType device\_type) | 创建一个运行时设备信息对象。 |
| void OH\_AI\_ContextDestroy(OH\_AI\_ContextHandle \*context) | 释放上下文对象。 |
| void OH\_AI\_ContextAddDeviceInfo(OH\_AI\_ContextHandle context, OH\_AI\_DeviceInfoHandle device\_info) | 添加运行时设备信息。 |
| OH\_AI\_TrainCfgHandle OH\_AI\_TrainCfgCreate() | 创建训练配置对象指针。 |
| void OH\_AI\_TrainCfgDestroy(OH\_AI\_TrainCfgHandle \*train\_cfg) | 销毁训练配置对象指针。 |
| OH\_AI\_ModelHandle OH\_AI\_ModelCreate() | 创建一个模型对象。 |
| OH\_AI\_Status OH\_AI\_TrainModelBuildFromFile(OH\_AI\_ModelHandle model, const char \*model\_path, OH\_AI\_ModelType model\_type, const OH\_AI\_ContextHandle model\_context, const OH\_AI\_TrainCfgHandle train\_cfg) | 通过模型文件加载并编译MindSpore Lite训练模型。 |
| OH\_AI\_Status OH\_AI\_RunStep(OH\_AI\_ModelHandle model, const OH\_AI\_KernelCallBack before, const OH\_AI\_KernelCallBack after) | 单步训练模型。 |
| OH\_AI\_Status OH\_AI\_ModelSetTrainMode(OH\_AI\_ModelHandle model, bool train) | 设置训练模式。 |
| OH\_AI\_Status OH\_AI\_ExportModel(OH\_AI\_ModelHandle model, OH\_AI\_ModelType model\_type, const char \*model\_file, OH\_AI\_QuantizationType quantization\_type, bool export\_inference\_only, char \*\*output\_tensor\_name, size\_t num) | 导出训练后的ms模型。 |
| void OH\_AI\_ModelDestroy(OH\_AI\_ModelHandle \*model) | 释放一个模型对象。 |

#### 开发步骤

使用MindSpore Lite进行模型训练的开发流程如下图所示。

**图 1** 使用MindSpore Lite进行模型训练的开发流程

![](https://contentcenter-vali-drcn.dbankcdn.cn/pvt_2/DeveloperAlliance_scene_100_1/3f/v3/Th0u1DVQQF2-X3AeDZvhPQ/zh-cn_image_0000002594829580.png?HW-CC-KV=V1&HW-CC-Date=20260612T073804Z&HW-CC-Expire=86400&HW-CC-Sign=2C2B446A9F8254A836AC1D5291A775D296953413B48A189E0AF632866258B7CD)

进入主要流程之前需要先引用相关的头文件，并编写函数生成随机的输入，具体如下：

#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include "mindspore/model.h"

int GenerateInputDataWithRandom(OH\_AI\_TensorHandleArray inputs) {
  for (size\_t i = 0; i < inputs.handle\_num; ++i) {
    float \*input\_data = (float \*)OH\_AI\_TensorGetMutableData(inputs.handle\_list\[i\]);
    if (input\_data == NULL) {
      printf("OH\_AI\_TensorGetMutableData failed.\\n");
      return  OH\_AI\_STATUS\_LITE\_ERROR;
    }
    int64\_t num = OH\_AI\_TensorGetElementNum(inputs.handle\_list\[i\]);
    const int divisor = 10;
    for (size\_t j = 0; j < num; j++) {
      input\_data\[j\] = (float)(rand() % divisor) / divisor;  // 0--0.9f
    }
  }
  return OH\_AI\_STATUS\_SUCCESS;
}

然后进入主要的开发步骤，包括模型的准备、读取、编译、训练、模型导出和释放，具体开发过程及细节请见下文的开发步骤及示例。

1.  模型准备。
    
    准备的模型格式为.ms，本文以lenet\_train.ms为例（此模型是提前准备的ms模型，本文相关效果仅以此模型文件为例）。开发者请自行准备所需的模型，可以按如下步骤操作：
    
    *   首先基于MindSpore架构使用Python创建网络模型，并导出为.mindir文件，详细指南参考[这里](https://www.mindspore.cn/tutorials/zh-CN/r2.1/beginner/quick_start.html)。
    *   然后将.mindir模型文件转换成.ms文件，转换操作步骤可以参考[训练模型转换](https://www.mindspore.cn/lite/docs/zh-CN/r2.1/use/converter_train.html)，.ms文件可以导入端侧设备并基于MindSpore Lite端侧框架进行训练。
2.  创建上下文，设置设备类型、训练配置等参数。
    
    // Create and init context, add CPU device info
    OH\_AI\_ContextHandle context = OH\_AI\_ContextCreate();
    if (context == NULL) {
        printf("OH\_AI\_ContextCreate failed.\\n");
        return OH\_AI\_STATUS\_LITE\_ERROR;
    }
    
    OH\_AI\_DeviceInfoHandle cpu\_device\_info = OH\_AI\_DeviceInfoCreate(OH\_AI\_DEVICETYPE\_CPU);
    if (cpu\_device\_info == NULL) {
        printf("OH\_AI\_DeviceInfoCreate failed.\\n");
        OH\_AI\_ContextDestroy(&context);
        return OH\_AI\_STATUS\_LITE\_ERROR;
    }
    OH\_AI\_ContextAddDeviceInfo(context, cpu\_device\_info);
    
    // Create trainCfg
    OH\_AI\_TrainCfgHandle trainCfg = OH\_AI\_TrainCfgCreate();
    if (trainCfg == NULL) {
        printf("OH\_AI\_TrainCfgCreate failed.\\n");
        OH\_AI\_ContextDestroy(&context);
        return OH\_AI\_STATUS\_LITE\_ERROR;
    }
    
3.  创建、加载与编译模型。
    
    调用OH\_AI\_TrainModelBuildFromFile加载并编译模型。
    
    // Create model
    OH\_AI\_ModelHandle model = OH\_AI\_ModelCreate();
    if (model == NULL) {
        printf("OH\_AI\_ModelCreate failed.\\n");
        OH\_AI\_TrainCfgDestroy(&trainCfg);
        OH\_AI\_ContextDestroy(&context);
        return OH\_AI\_STATUS\_LITE\_ERROR;
    }
    
    // Build model
    int ret = OH\_AI\_TrainModelBuildFromFile(model, model\_file, OH\_AI\_MODELTYPE\_MINDIR, context, trainCfg);
    if (ret != OH\_AI\_STATUS\_SUCCESS) {
        printf("OH\_AI\_TrainModelBuildFromFile failed, ret: %d.\\n", ret);
        OH\_AI\_ModelDestroy(&model);
        OH\_AI\_ContextDestroy(&context);
        return ret;
    }
    
4.  输入数据。
    
    模型执行之前需要向输入的张量中填充数据。本例使用随机的数据对模型进行填充。
    
    // Get Inputs
    OH\_AI\_TensorHandleArray inputs = OH\_AI\_ModelGetInputs(model);
    if (inputs.handle\_list == NULL) {
        printf("OH\_AI\_ModelGetInputs failed, ret: %d.\\n", ret);
        OH\_AI\_ModelDestroy(&model);
        OH\_AI\_ContextDestroy(&context);
        return ret;
    }
    
    // Generate random data as input data.
    ret = GenerateInputDataWithRandom(inputs);
    if (ret != OH\_AI\_STATUS\_SUCCESS) {
        printf("GenerateInputDataWithRandom failed, ret: %d.\\n", ret);
        OH\_AI\_ModelDestroy(&model);
        OH\_AI\_ContextDestroy(&context);
        return ret;
    }
    
5.  执行训练。
    
    使用OH\_AI\_ModelSetTrainMode接口设置训练模式，使用OH\_AI\_RunStep接口进行模型训练。
    
    // Set Train Mode
    ret = OH\_AI\_ModelSetTrainMode(model, true);
    if (ret != OH\_AI\_STATUS\_SUCCESS) {
        printf("OH\_AI\_ModelSetTrainMode failed, ret: %d.\\n", ret);
        OH\_AI\_ModelDestroy(&model);
        OH\_AI\_ContextDestroy(&context);
        return ret;
    }
    
    // Model Train Step
    ret = OH\_AI\_RunStep(model, NULL, NULL);
    if (ret != OH\_AI\_STATUS\_SUCCESS) {
        printf("OH\_AI\_RunStep failed, ret: %d.\\n", ret);
        OH\_AI\_ModelDestroy(&model);
        OH\_AI\_ContextDestroy(&context);
        return ret;
    }
    printf("Train Step Success.\\n");
    
6.  导出训练后模型。
    
    使用OH\_AI\_ExportModel接口导出训练后模型。
    
    // Export Train Model
    ret = OH\_AI\_ExportModel(model, OH\_AI\_MODELTYPE\_MINDIR, export\_train\_model, OH\_AI\_NO\_QUANT, false, NULL, 0);
    if (ret != OH\_AI\_STATUS\_SUCCESS) {
        printf("OH\_AI\_ExportModel train failed, ret: %d.\\n", ret);
        OH\_AI\_ModelDestroy(&model);
        OH\_AI\_ContextDestroy(&context);
        return ret;
    }
    printf("Export Train Model Success.\\n");
    
    // Export Inference Model
    ret = OH\_AI\_ExportModel(model, OH\_AI\_MODELTYPE\_MINDIR, export\_infer\_model, OH\_AI\_NO\_QUANT, true, NULL, 0);
    if (ret != OH\_AI\_STATUS\_SUCCESS) {
        printf("OH\_AI\_ExportModel inference failed, ret: %d.\\n", ret);
        OH\_AI\_ModelDestroy(&model);
        OH\_AI\_ContextDestroy(&context);
        return ret;
    }
    printf("Export Inference Model Success.\\n");
    
7.  释放模型。
    
    不再使用MindSpore Lite推理框架时，需要释放已经创建的模型。
    
    // Delete model and context.
    OH\_AI\_ModelDestroy(&model);
    OH\_AI\_ContextDestroy(&context);
    

#### 调测验证

1.  编写CMakeLists.txt。
    
    cmake\_minimum\_required(VERSION 3.14)
    project(TrainDemo)
    
    add\_executable(train\_demo main.c)
    
    target\_link\_libraries(
            train\_demo
            mindspore\_lite\_ndk
    )
    
    *   使用ohos-sdk交叉编译，需要对CMake设置native工具链路径，即：-DCMAKE\_TOOLCHAIN\_FILE="/xxx/native/build/cmake/ohos.toolchain.cmake"。
        
    *   编译命令如下，其中OHOS\_NDK需要设置为native工具链路径：
        
          mkdir -p build
        
          cd ./build || exit
          OHOS\_NDK=""
          cmake -G "Unix Makefiles" \\
                -S ../ \\
                -DCMAKE\_TOOLCHAIN\_FILE="$OHOS\_NDK/build/cmake/ohos.toolchain.cmake" \\
                -DOHOS\_ARCH=arm64-v8a \\
                -DCMAKE\_BUILD\_TYPE=Release
        
          make
        
2.  运行编译的可执行程序。
    
    *   使用hdc连接设备，并将train\_demo和lenet\_train.ms推送到设备中的相同目录。
    *   使用hdc shell进入设备，并进入train\_demo所在的目录执行如下命令，即可得到结果。
    
    ./train\_demo ./lenet\_train.ms export\_train\_model export\_infer\_model
    
    得到如下输出：
    
    Train Step Success.
    Export Train Model Success.
    Export Inference Model Success.
    Tensor name: Default/network-WithLossCell/\_backbone-LeNet5/fc3-Dense/BiasAdd-op121, tensor size is 80, elements num: 20.
    output data is:
    0.000265 0.000231 0.000254 0.000269 0.000238 0.000228
    
    在train\_demo所在目录可以看到导出的两个模型文件：export\_train\_model.ms和export\_infer\_model.ms。
    

#### 完整示例

#include <stdlib.h>
#include <stdio.h>
#include <string>
#include "mindspore/model.h"

int GenerateInputDataWithRandom(OH\_AI\_TensorHandleArray inputs) {
  for (size\_t i = 0; i < inputs.handle\_num; ++i) {
    float \*input\_data = (float \*)OH\_AI\_TensorGetMutableData(inputs.handle\_list\[i\]);
    if (input\_data == NULL) {
      printf("OH\_AI\_TensorGetMutableData failed.\\n");
      return  OH\_AI\_STATUS\_LITE\_ERROR;
    }
    int64\_t num = OH\_AI\_TensorGetElementNum(inputs.handle\_list\[i\]);
    const int divisor = 10;
    for (size\_t j = 0; j < num; j++) {
      input\_data\[j\] = (float)(rand() % divisor) / divisor;  // 0--0.9f
    }
  }
  return OH\_AI\_STATUS\_SUCCESS;
}

int ModelPredict(char\* model\_file) {
  // Create and init context, add CPU device info
  OH\_AI\_ContextHandle context = OH\_AI\_ContextCreate();
  if (context == NULL) {
    printf("OH\_AI\_ContextCreate failed.\\n");
    return OH\_AI\_STATUS\_LITE\_ERROR;
  }

  OH\_AI\_DeviceInfoHandle cpu\_device\_info = OH\_AI\_DeviceInfoCreate(OH\_AI\_DEVICETYPE\_CPU);
  if (cpu\_device\_info == NULL) {
    printf("OH\_AI\_DeviceInfoCreate failed.\\n");
    OH\_AI\_ContextDestroy(&context);
    return OH\_AI\_STATUS\_LITE\_ERROR;
  }
  OH\_AI\_ContextAddDeviceInfo(context, cpu\_device\_info);

  // Create model
  OH\_AI\_ModelHandle model = OH\_AI\_ModelCreate();
  if (model == NULL) {
    printf("OH\_AI\_ModelCreate failed.\\n");
    OH\_AI\_ContextDestroy(&context);
    return OH\_AI\_STATUS\_LITE\_ERROR;
  }

  // Build model
  int ret = OH\_AI\_ModelBuildFromFile(model, model\_file, OH\_AI\_MODELTYPE\_MINDIR, context);
  if (ret != OH\_AI\_STATUS\_SUCCESS) {
    printf("OH\_AI\_ModelBuildFromFile failed, ret: %d.\\n", ret);
    OH\_AI\_ModelDestroy(&model);
    OH\_AI\_ContextDestroy(&context);
    return ret;
  }

  // Get Inputs
  OH\_AI\_TensorHandleArray inputs = OH\_AI\_ModelGetInputs(model);
  if (inputs.handle\_list == NULL) {
    printf("OH\_AI\_ModelGetInputs failed, ret: %d.\\n", ret);
    OH\_AI\_ModelDestroy(&model);
    OH\_AI\_ContextDestroy(&context);
    return ret;
  }

  // Generate random data as input data.
  ret = GenerateInputDataWithRandom(inputs);
  if (ret != OH\_AI\_STATUS\_SUCCESS) {
    printf("GenerateInputDataWithRandom failed, ret: %d.\\n", ret);
    OH\_AI\_ModelDestroy(&model);
    OH\_AI\_ContextDestroy(&context);
    return ret;
  }

  // Model Predict
  OH\_AI\_TensorHandleArray outputs;
  ret = OH\_AI\_ModelPredict(model, inputs, &outputs, NULL, NULL);
  if (ret != OH\_AI\_STATUS\_SUCCESS) {
    printf("MSModelPredict failed, ret: %d.\\n", ret);
    OH\_AI\_ModelDestroy(&model);
    OH\_AI\_ContextDestroy(&context);
    return ret;
  }

  // Print Output Tensor Data.
  for (size\_t i = 0; i < outputs.handle\_num; ++i) {
    OH\_AI\_TensorHandle tensor = outputs.handle\_list\[i\];
    int64\_t element\_num = OH\_AI\_TensorGetElementNum(tensor);
    printf("Tensor name: %s, tensor size is %ld ,elements num: %ld.\\n", OH\_AI\_TensorGetName(tensor),
           OH\_AI\_TensorGetDataSize(tensor), element\_num);
    const float \*data = (const float \*)OH\_AI\_TensorGetData(tensor);
    printf("output data is:\\n");
    const int max\_print\_num = 50;
    for (int j = 0; j < element\_num && j <= max\_print\_num; ++j) {
      printf("%f ", data\[j\]);
    }
    printf("\\n");
  }

  OH\_AI\_ModelDestroy(&model);
  OH\_AI\_ContextDestroy(&context);
  return OH\_AI\_STATUS\_SUCCESS;
}

int TrainDemo(int argc, const char \*\*argv) {
  if (argc < 4) {
    printf("Model file must be provided.\\n");
    printf("Export Train Model path must be provided.\\n");
    printf("Export Inference Model path must be provided.\\n");
    return OH\_AI\_STATUS\_LITE\_ERROR;
  }
  const char \*model\_file = argv\[1\];
  const char \*export\_train\_model = argv\[2\];
  const char \*export\_infer\_model = argv\[3\];

  // Create and init context, add CPU device info
  OH\_AI\_ContextHandle context = OH\_AI\_ContextCreate();
  if (context == NULL) {
    printf("OH\_AI\_ContextCreate failed.\\n");
    return OH\_AI\_STATUS\_LITE\_ERROR;
  }

  OH\_AI\_DeviceInfoHandle cpu\_device\_info = OH\_AI\_DeviceInfoCreate(OH\_AI\_DEVICETYPE\_CPU);
  if (cpu\_device\_info == NULL) {
    printf("OH\_AI\_DeviceInfoCreate failed.\\n");
    OH\_AI\_ContextDestroy(&context);
    return OH\_AI\_STATUS\_LITE\_ERROR;
  }
  OH\_AI\_ContextAddDeviceInfo(context, cpu\_device\_info);

  // Create trainCfg
  OH\_AI\_TrainCfgHandle trainCfg = OH\_AI\_TrainCfgCreate();
  if (trainCfg == NULL) {
    printf("OH\_AI\_TrainCfgCreate failed.\\n");
    OH\_AI\_ContextDestroy(&context);
    return OH\_AI\_STATUS\_LITE\_ERROR;
  }

  // Create model
  OH\_AI\_ModelHandle model = OH\_AI\_ModelCreate();
  if (model == NULL) {
    printf("OH\_AI\_ModelCreate failed.\\n");
    OH\_AI\_TrainCfgDestroy(&trainCfg);
    OH\_AI\_ContextDestroy(&context);
    return OH\_AI\_STATUS\_LITE\_ERROR;
  }

  // Build model
  int ret = OH\_AI\_TrainModelBuildFromFile(model, model\_file, OH\_AI\_MODELTYPE\_MINDIR, context, trainCfg);
  if (ret != OH\_AI\_STATUS\_SUCCESS) {
    printf("OH\_AI\_TrainModelBuildFromFile failed, ret: %d.\\n", ret);
    OH\_AI\_ModelDestroy(&model);
    OH\_AI\_ContextDestroy(&context);
    return ret;
  }

  // Get Inputs
  OH\_AI\_TensorHandleArray inputs = OH\_AI\_ModelGetInputs(model);
  if (inputs.handle\_list == NULL) {
    printf("OH\_AI\_ModelGetInputs failed, ret: %d.\\n", ret);
    OH\_AI\_ModelDestroy(&model);
    OH\_AI\_ContextDestroy(&context);
    return ret;
  }

  // Generate random data as input data.
  ret = GenerateInputDataWithRandom(inputs);
  if (ret != OH\_AI\_STATUS\_SUCCESS) {
    printf("GenerateInputDataWithRandom failed, ret: %d.\\n", ret);
    OH\_AI\_ModelDestroy(&model);
    OH\_AI\_ContextDestroy(&context);
    return ret;
  }

  // Set Train Mode
  ret = OH\_AI\_ModelSetTrainMode(model, true);
  if (ret != OH\_AI\_STATUS\_SUCCESS) {
    printf("OH\_AI\_ModelSetTrainMode failed, ret: %d.\\n", ret);
    OH\_AI\_ModelDestroy(&model);
    OH\_AI\_ContextDestroy(&context);
    return ret;
  }

  // Model Train Step
  ret = OH\_AI\_RunStep(model, NULL, NULL);
  if (ret != OH\_AI\_STATUS\_SUCCESS) {
    printf("OH\_AI\_RunStep failed, ret: %d.\\n", ret);
    OH\_AI\_ModelDestroy(&model);
    OH\_AI\_ContextDestroy(&context);
    return ret;
  }
  printf("Train Step Success.\\n");

  // Export Train Model
  ret = OH\_AI\_ExportModel(model, OH\_AI\_MODELTYPE\_MINDIR, export\_train\_model, OH\_AI\_NO\_QUANT, false, NULL, 0);
  if (ret != OH\_AI\_STATUS\_SUCCESS) {
    printf("OH\_AI\_ExportModel train failed, ret: %d.\\n", ret);
    OH\_AI\_ModelDestroy(&model);
    OH\_AI\_ContextDestroy(&context);
    return ret;
  }
  printf("Export Train Model Success.\\n");

  // Export Inference Model
  ret = OH\_AI\_ExportModel(model, OH\_AI\_MODELTYPE\_MINDIR, export\_infer\_model, OH\_AI\_NO\_QUANT, true, NULL, 0);
  if (ret != OH\_AI\_STATUS\_SUCCESS) {
    printf("OH\_AI\_ExportModel inference failed, ret: %d.\\n", ret);
    OH\_AI\_ModelDestroy(&model);
    OH\_AI\_ContextDestroy(&context);
    return ret;
  }
  printf("Export Inference Model Success.\\n");

  // Delete model and context.
  OH\_AI\_ModelDestroy(&model);
  OH\_AI\_ContextDestroy(&context);

  // Use The Exported Model to predict
  std::string temp\_path = std::string(export\_infer\_model) + ".ms";
  const char \*exported\_model = temp\_path.c\_str();
  ret = ModelPredict(exported\_model);
  if (ret != OH\_AI\_STATUS\_SUCCESS) {
    printf("Exported Model to predict failed, ret: %d.\\n", ret);
    return ret;
  }
  return OH\_AI\_STATUS\_SUCCESS;
}

int main(int argc, const char \*\*argv) { return TrainDemo(argc, argv); }