diff --git a/model_zoo/official/cv/lenet/scripts/run_standalone_eval_cpu.sh b/model_zoo/official/cv/lenet/scripts/run_standalone_eval_cpu.sh index e05dbc99652..10b9f0d836b 100755 --- a/model_zoo/official/cv/lenet/scripts/run_standalone_eval_cpu.sh +++ b/model_zoo/official/cv/lenet/scripts/run_standalone_eval_cpu.sh @@ -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 & diff --git a/model_zoo/official/cv/lenet/scripts/run_standalone_train_cpu.sh b/model_zoo/official/cv/lenet/scripts/run_standalone_train_cpu.sh index f494467a7c9..ced49d77ec3 100755 --- a/model_zoo/official/cv/lenet/scripts/run_standalone_train_cpu.sh +++ b/model_zoo/official/cv/lenet/scripts/run_standalone_train_cpu.sh @@ -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 &