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

fix lenet performance on cpu

This commit is contained in:
baihuawei 2021-04-14 15:36:06 +08:00
parent 7211f6f1f4
commit 1b027c11bf
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 &