From a63a3947b0f5f4403dfbba71c9f0e0bcd25fd99b Mon Sep 17 00:00:00 2001 From: MRXLT Date: Wed, 1 Apr 2020 09:03:02 +0000 Subject: [PATCH] fix bert benchmark script --- python/examples/bert/benchmark_batch.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/python/examples/bert/benchmark_batch.py b/python/examples/bert/benchmark_batch.py index 9b8e301a..7cedb6aa 100644 --- a/python/examples/bert/benchmark_batch.py +++ b/python/examples/bert/benchmark_batch.py @@ -53,7 +53,7 @@ def single_func(idx, resource): feed_batch.append(reader.process(dataset[bi])) b_end = time.time() if profile_flags: - print("PROFILE\tpid:{}\tbert+pre_0:{} bert_pre_1:{}".format( + print("PROFILE\tpid:{}\tbert_pre_0:{} bert_pre_1:{}".format( os.getpid(), int(round(b_start * 1000000)), int(round(b_end * 1000000)))) @@ -69,9 +69,7 @@ def single_func(idx, resource): if __name__ == '__main__': multi_thread_runner = MultiThreadRunner() - endpoint_list = [ - "127.0.0.1:9292", "127.0.0.1:9293", "127.0.0.1:9294", "127.0.0.1:9295" - ] + endpoint_list = ["127.0.0.1:9292"] result = multi_thread_runner.run(single_func, args.thread, {"endpoint": endpoint_list}) avg_cost = 0 -- GitLab