benchmark.sh 622 字节
Newer Older
W
wangjiawei04 已提交
1
rm profile_log
M
MRXLT 已提交
2 3
export FLAGS_profile_client=1
export FLAGS_profile_server=1
W
wangjiawei04 已提交
4 5
for thread_num in 1 2 4 8 16
do
M
MRXLT 已提交
6
for batch_size in 1 4 16 64 256
M
MRXLT 已提交
7 8
do
    $PYTHONROOT/bin/python benchmark.py --thread $thread_num --batch_size $batch_size --model serving_client_conf/serving_client_conf.prototxt --request rpc > profile 2>&1
M
MRXLT 已提交
9 10
    echo "batch size : $batch_size"
    echo "thread num : $thread_num"
W
wangjiawei04 已提交
11 12 13 14 15
    echo "========================================"
    echo "batch size : $batch_size" >> profile_log
    $PYTHONROOT/bin/python ../util/show_profile.py profile $thread_num >> profile_log
    tail -n 2 profile >> profile_log
done
M
MRXLT 已提交
16
done