From 298bb59033d5a21f498a97b3539cba53ba1420a4 Mon Sep 17 00:00:00 2001 From: MRXLT Date: Thu, 16 Jan 2020 06:24:14 +0000 Subject: [PATCH] add readme for imdb examples --- core/general-client/src/general_model.cpp | 2 +- python/examples/imdb/README.md | 5 ++++- python/paddle_serving/serving_client/__init__.py | 8 ++++++-- 3 files changed, 11 insertions(+), 4 deletions(-) diff --git a/core/general-client/src/general_model.cpp b/core/general-client/src/general_model.cpp index 42ac29ee..4c22fa48 100644 --- a/core/general-client/src/general_model.cpp +++ b/core/general-client/src/general_model.cpp @@ -254,7 +254,7 @@ std::vector>> PredictorClient::predict_for_batch( fetch_result_batch[bi][idx].resize(len); for (int i = 0; i < len; ++i) { fetch_result_batch[bi][idx][i] = - *(const float *)res.insts(0).tensor_array(idx).data(i).c_str(); + *(const float *)res.insts(bi).tensor_array(idx).data(i).c_str(); } } } diff --git a/python/examples/imdb/README.md b/python/examples/imdb/README.md index 9106dd87..afe18df8 100644 --- a/python/examples/imdb/README.md +++ b/python/examples/imdb/README.md @@ -10,4 +10,7 @@ cat test.data | python test_client.py > result ``` python test_client_multithread.py inference.conf test.data 4 > result ``` -batch clienit +batch clienit，若batch size为4 +``` +cat test.data | python test_client_batch inference.conf 4 > result +``` diff --git a/python/paddle_serving/serving_client/__init__.py b/python/paddle_serving/serving_client/__init__.py index 8bf66904..200b99e4 100644 --- a/python/paddle_serving/serving_client/__init__.py +++ b/python/paddle_serving/serving_client/__init__.py @@ -161,6 +161,7 @@ class Client(object): int_feed_names = [] float_feed_names = [] fetch_names = [] + counter = 0 for feed in feed_batch: int_slot = [] float_slot = [] @@ -168,11 +169,14 @@ class Client(object): if key not in self.feed_names_: continue if self.feed_types_[key] == int_type: - int_feed_names.append(key) + if counter == 0: + int_feed_names.append(key) int_slot.append(feed[key]) elif self.feed_types_[key] == float_type: - float_feed_names.append(key) + if counter == 0: + float_feed_names.append(key) float_slot.append(feed[key]) + counter += 1 int_slot_batch.append(int_slot) float_slot_batch.append(float_slot) -- GitLab