Fix reader creator in mnist

8 years ago · 01ad3bfdb9
parent e55ec44b37
commit 01ad3bfdb9
2 changed files with 2 additions and 2 deletions
--- a/demo/mnist/api_train_v2.py
+++ b/demo/mnist/api_train_v2.py
@ -33,7 +33,7 @@ def main():
    trainer.train(
        reader=paddle.reader.batched(
            paddle.reader.shuffle(
-                paddle.dataset.mnist.train_creator(), buf_size=8192),
+                paddle.dataset.mnist.train(), buf_size=8192),
            batch_size=32),
        cost=cost,
        parameters=parameters,
--- a/python/paddle/v2/dataset/mnist.py
+++ b/python/paddle/v2/dataset/mnist.py
@ -45,7 +45,7 @@ def reader_creator(image_filename, label_filename, buffer_size):
        m.terminate()
        l.terminate()
-    return reader()
+    return reader
 def train():