提交 86b8bdc0 编写于 作者: T tensor-tang

enable inference alexnet benchmark

上级 19939657
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册