enable inference alexnet benchmark

del_some_in_makelist
tensor-tang 7 years ago
parent 19939657ea
commit 86b8bdc0af

@ -7,8 +7,17 @@ width = 227
num_class = 1000
batch_size = get_config_arg('batch_size', int, 128)
use_mkldnn = get_config_arg('use_mkldnn', bool, False)
is_infer = get_config_arg("is_infer", bool, False)
num_samples = get_config_arg('num_samples', int, 2560)
args = {'height': height, 'width': width, 'color': True, 'num_class': num_class}
args = {
'height': height,
'width': width,
'color': True,
'num_class': num_class,
'is_infer': is_infer,
'num_samples': num_samples
}
define_py_data_sources2(
"train.list", None, module="provider", obj="process", args=args)

@ -79,6 +79,7 @@ fi
# inference benchmark
for use_mkldnn in True False; do
for batchsize in 1 2 4 8 16; do
infer alexnet group2 $batchsize $use_mkldnn
infer googlenet v1 $batchsize $use_mkldnn
infer resnet 50 $batchsize $use_mkldnn
infer vgg 19 $batchsize $use_mkldnn

@ -56,6 +56,7 @@ fi
# inference benchmark
for batchsize in 1 2 4 8 16; do
infer alexnet group2 $batchsize $use_mkldnn
infer googlenet v1 $batchsize
infer resnet 50 $batchsize
infer vgg 19 $batchsize

Loading…
Cancel
Save