From b95365bee5db5952b971465f9fc867b22ffdd462 Mon Sep 17 00:00:00 2001 From: MRXLT Date: Fri, 20 Mar 2020 10:32:25 +0800 Subject: [PATCH] fix bug for batch predict test=serving --- python/paddle_serving_client/__init__.py | 21 +++++++++++---------- 1 file changed, 11 insertions(+), 10 deletions(-) diff --git a/python/paddle_serving_client/__init__.py b/python/paddle_serving_client/__init__.py index c77034b5..ce0eb8c8 100644 --- a/python/paddle_serving_client/__init__.py +++ b/python/paddle_serving_client/__init__.py @@ -228,16 +228,17 @@ class Client(object): fetch_names, result_batch, self.pid) result_map_batch = [] - for index in range(batch_size): - result_map = {} - for i, name in enumerate(fetch_names): - if self.fetch_names_to_type_[name] == int_type: - result_map[name] = result_batch.get_int64_by_name(name)[ - index] - elif self.fetch_names_to_type_[name] == float_type: - result_map[name] = result_batch.get_float_by_name(name)[ - index] - result_map_batch.append(result_map) + result_map = {} + for i, name in enumerate(fetch_names): + if self.fetch_names_to_type_[name] == int_type: + result_map[name] = result_batch.get_int64_by_name(name) + elif self.fetch_names_to_type_[name] == float_type: + result_map[name] = result_batch.get_float_by_name(name) + for i in range(batch_size): + single_result = {} + for key in result_map: + single_result[key] = result_map[key][i] + result_map_batch.append(single_result) return result_map_batch -- GitLab