diff --git a/benchmark/paddle/image/run.mkldnn.sh b/benchmark/paddle/image/run.mkldnn.sh index 03a87afbc3044a6d3e65d1fb87d5d4e8d3a5e8b1..9f7624369590097e362c94a809db92d8962089b7 100755 --- a/benchmark/paddle/image/run.mkldnn.sh +++ b/benchmark/paddle/image/run.mkldnn.sh @@ -1,5 +1,9 @@ set -e +unset OMP_NUM_THREADS MKL_NUM_THREADS +export OMP_DYNAMIC="FALSE" +export KMP_AFFINITY="granularity=fine,compact,0,0" + function train() { topology=$1 bs=$2 diff --git a/benchmark/paddle/image/vgg.py b/benchmark/paddle/image/vgg.py index 69e4a4cddb47e9402e00b452317e11c6051362c7..b8429975f5c83df6996e71478fe276b246e8b77b 100644 --- a/benchmark/paddle/image/vgg.py +++ b/benchmark/paddle/image/vgg.py @@ -5,7 +5,7 @@ height = 224 width = 224 num_class = 1000 batch_size = get_config_arg('batch_size', int, 64) -layer_num = get_config_arg('layer_num', int, 16) +layer_num = get_config_arg('layer_num', int, 19) args = {'height': height, 'width': width, 'color': True, 'num_class': num_class} define_py_data_sources2(