p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!15168 fix lenet performance on cpu 

From: @huaweib
Reviewed-by: @jjfeing,@kisnwang
Signed-off-by: @kisnwang
This commit is contained in:
mindspore-ci-bot 2021-04-15 10:47:00 +08:00 committed by Gitee
parent f137406364 1b027c11bf
commit 8af28def00
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
model_zoo/official/cv/lenet/scripts/run_standalone_eval_cpu.sh
View File

@ -19,4 +19,5 @@ script_self=$(readlink -f "$0")
self_path=$(dirname "${script_self}")
DATA_PATH=$1
CKPT_PATH=$2
export OMP_NUM_THREADS=1
python -s ${self_path}/../eval.py --data_path=$DATA_PATH --device_target="CPU" --ckpt_path=$CKPT_PATH > log.txt 2>&1 &

1
model_zoo/official/cv/lenet/scripts/run_standalone_train_cpu.sh
View File

@ -19,4 +19,5 @@ script_self=$(readlink -f "$0")
self_path=$(dirname "${script_self}")
DATA_PATH=$1
CKPT_PATH=$2
export OMP_NUM_THREADS=1
python -s ${self_path}/../train.py --data_path=$DATA_PATH --device_target="CPU" --ckpt_path=$CKPT_PATH > log.txt 2>&1 &