Follow comments

9 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,9 +43,6 @@ class Inference(object):
            def __reader_impl__():
                for each_sample in input:
                    if len(reader_dict) == 1:
                        yield [each_sample]
                    else:
                    yield each_sample
            reader = minibatch.batch(__reader_impl__, batch_size=batch_size)