Follow comments

8 years ago · 0d2d419a59
parent d5365bb715
commit 0d2d419a59
2 changed files with 2 additions and 5 deletions
--- a/demo/mnist/api_train_v2.py
+++ b/demo/mnist/api_train_v2.py
@ -113,7 +113,7 @@ def main():
    test_creator = paddle.dataset.mnist.test()
    test_data = []
    for item in test_creator():
-        test_data.append(item[0])
+        test_data.append((item[0], ))
        if len(test_data) == 100:
            break

--- a/python/paddle/v2/inference.py
+++ b/python/paddle/v2/inference.py
@ -43,10 +43,7 @@ class Inference(object):

            def __reader_impl__():
                for each_sample in input:
-                    if len(reader_dict) == 1:
-                        yield [each_sample]
-                    else:
-                        yield each_sample
+                    yield each_sample

            reader = minibatch.batch(__reader_impl__, batch_size=batch_size)
        else: