Huawei_Technology
/
mindspore
Not watched
1
0
Fork 0
Code
Releases 13 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
Browse Source

fix lenet performance on cpu

pull/15168/head
baihuawei 4 years ago
parent
7211f6f1f4
commit
1b027c11bf
2 changed files with 2 additions and 0 deletions
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +1
    -0
      model_zoo/official/cv/lenet/scripts/run_standalone_eval_cpu.sh
  2. +1
    -0
      model_zoo/official/cv/lenet/scripts/run_standalone_train_cpu.sh

+ 1
- 0
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}") self_path=$(dirname "${script_self}")
DATA_PATH=$1 DATA_PATH=$1
CKPT_PATH=$2 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 & python -s ${self_path}/../eval.py --data_path=$DATA_PATH --device_target="CPU" --ckpt_path=$CKPT_PATH > log.txt 2>&1 &

+ 1
- 0
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}") self_path=$(dirname "${script_self}")
DATA_PATH=$1 DATA_PATH=$1
CKPT_PATH=$2 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 & python -s ${self_path}/../train.py --data_path=$DATA_PATH --device_target="CPU" --ckpt_path=$CKPT_PATH > log.txt 2>&1 &

Write Preview
Loading…
Cancel
Save