提交 9928eb81 编写于 作者: T Tao Luo 提交者: GitHub

Merge pull request #4427 from tensor-tang/mkldnn

refine mkldnn benchmark script
...@@ -13,9 +13,13 @@ function train() { ...@@ -13,9 +13,13 @@ function train() {
log="logs/${topology}-mkldnn-${bs}.log" log="logs/${topology}-mkldnn-${bs}.log"
elif [ $3 == "False" ]; then elif [ $3 == "False" ]; then
thread=`nproc` thread=`nproc`
# each trainer_count use only 1 core to avoid conflict
export OMP_NUM_THREADS=1
export MKL_NUM_THREADS=1
log="logs/${topology}-${thread}mklml-${bs}.log" log="logs/${topology}-${thread}mklml-${bs}.log"
else else
echo "Wrong input $3, use True or False." echo "Wrong input $3, use True or False."
exit 0
fi fi
args="batch_size=${bs}" args="batch_size=${bs}"
config="${topology}.py" config="${topology}.py"
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册