From b9f5224d5da9d8ae5b956a00d7bb00ad2e5d1261 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E4=B8=87=E7=A5=96=E6=B6=9B?= <1025494833@qq.com> Date: Tue, 6 Feb 2024 12:25:27 +0000 Subject: [PATCH 1/2] =?UTF-8?q?!10=E4=BF=AE=E6=94=B9HandWritingTrainAndInf?= =?UTF-8?q?er=E6=A0=B7=E4=BE=8BReadme?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: 万祖涛 <1025494833@qq.com> --- Samples/HandWritingTrainAndInfer/README.md | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/Samples/HandWritingTrainAndInfer/README.md b/Samples/HandWritingTrainAndInfer/README.md index 9acfa3d..657b124 100644 --- a/Samples/HandWritingTrainAndInfer/README.md +++ b/Samples/HandWritingTrainAndInfer/README.md @@ -64,8 +64,12 @@ #### 模型训练 1. 以HwHiAiUser用户登录开发板,切换到样例目录下。 - -2. 运行训练脚本。 +2. 设置环境变量减小算子编译内存占用。 + ``` + export TE_PARALLEL_COMPILER=1 + export MAX_COMPILE_CORE_NUMBER=1 + ``` +3. 运行训练脚本。 ``` python3 main.py -- Gitee From 8d1a7cc10fd34d86237605c0fabf3b076f6d6a53 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E4=B8=87=E7=A5=96=E6=B6=9B?= <1025494833@qq.com> Date: Wed, 7 Feb 2024 07:27:20 +0000 Subject: [PATCH 2/2] update Samples/HandWritingTrainAndInfer/README.md. MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: 万祖涛 <1025494833@qq.com> --- Samples/HandWritingTrainAndInfer/README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Samples/HandWritingTrainAndInfer/README.md b/Samples/HandWritingTrainAndInfer/README.md index 657b124..a802a56 100644 --- a/Samples/HandWritingTrainAndInfer/README.md +++ b/Samples/HandWritingTrainAndInfer/README.md @@ -2,7 +2,7 @@ #### 样例介绍 - +本样例使用MNIST数据集实现了手写数字识别体的训练,onnx推理,om推理全过程。 #### 样例下载 -- Gitee