Skip to content

Commit cc7b4b9

Browse files
authored
add return_numpy back (#10892)
1 parent 8653cf3 commit cc7b4b9

File tree

4 files changed

+9
-7
lines changed

4 files changed

+9
-7
lines changed

python/paddle/fluid/inferencer.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ def __init__(self, infer_func, param_path, place=None, parallel=False):
5656
else:
5757
self.exe = executor.Executor(self.place)
5858

59-
def infer(self, inputs):
59+
def infer(self, inputs, return_numpy=True):
6060
"""
6161
:param inputs: a map of {"input_name": input_var} that will be feed into the inference program
6262
to get the predict value
@@ -66,9 +66,11 @@ def infer(self, inputs):
6666
raise ValueError(
6767
"inputs should be a map of {'input_name': input_var}")
6868

69-
with self._prog_and_scope_guard():
70-
results = self.exe.run(feed=inputs,
71-
fetch_list=[self.predict_var.name])
69+
with executor.scope_guard(self.scope):
70+
results = self.exe.run(self.inference_program,
71+
feed=inputs,
72+
fetch_list=[self.predict_var],
73+
return_numpy=return_numpy)
7274

7375
return results
7476

python/paddle/fluid/tests/book/high-level-api/fit_a_line/test_fit_a_line.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ def infer(use_cuda, inference_program, save_dirname=None):
9292
tensor_x = numpy.random.uniform(0, 10, [batch_size, 13]).astype("float32")
9393

9494
results = inferencer.infer({'x': tensor_x})
95-
print("infer results: ", numpy.array(results[0]))
95+
print("infer results: ", results[0])
9696

9797

9898
def main(use_cuda):

python/paddle/fluid/tests/book/high-level-api/recognize_digits/test_recognize_digits_conv.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,7 @@ def infer(use_cuda, inference_program, save_dirname=None):
112112

113113
results = inferencer.infer({'img': tensor_img})
114114

115-
print("infer results: ", numpy.array(results[0]))
115+
print("infer results: ", results[0])
116116

117117

118118
def main(use_cuda):

python/paddle/fluid/tests/book/high-level-api/recognize_digits/test_recognize_digits_mlp.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,7 @@ def infer(use_cuda, inference_program, save_dirname=None):
9393

9494
results = inferencer.infer({'img': tensor_img})
9595

96-
print("infer results: ", numpy.array(results[0]))
96+
print("infer results: ", results[0])
9797

9898

9999
def main(use_cuda):

0 commit comments

Comments
 (0)