diff --git a/Samples/HandWritingTrainAndInfer/README.md b/Samples/HandWritingTrainAndInfer/README.md index 9acfa3d9fe559e64becf70d01e5d4774025119fa..a802a561061b0d35861f0d14fc25bd49391327ba 100644 --- a/Samples/HandWritingTrainAndInfer/README.md +++ b/Samples/HandWritingTrainAndInfer/README.md @@ -2,7 +2,7 @@ #### 样例介绍 - +本样例使用MNIST数据集实现了手写数字识别体的训练,onnx推理,om推理全过程。 #### 样例下载 @@ -64,8 +64,12 @@ #### 模型训练 1. 以HwHiAiUser用户登录开发板,切换到样例目录下。 - -2. 运行训练脚本。 +2. 设置环境变量减小算子编译内存占用。 + ``` + export TE_PARALLEL_COMPILER=1 + export MAX_COMPILE_CORE_NUMBER=1 + ``` +3. 运行训练脚本。 ``` python3 main.py