提交 01ad3bfd 编写于 作者: Y Yu Yang

Fix reader creator in mnist

上级 e55ec44b
...@@ -33,7 +33,7 @@ def main(): ...@@ -33,7 +33,7 @@ def main():
trainer.train( trainer.train(
reader=paddle.reader.batched( reader=paddle.reader.batched(
paddle.reader.shuffle( paddle.reader.shuffle(
paddle.dataset.mnist.train_creator(), buf_size=8192), paddle.dataset.mnist.train(), buf_size=8192),
batch_size=32), batch_size=32),
cost=cost, cost=cost,
parameters=parameters, parameters=parameters,
......
...@@ -45,7 +45,7 @@ def reader_creator(image_filename, label_filename, buffer_size): ...@@ -45,7 +45,7 @@ def reader_creator(image_filename, label_filename, buffer_size):
m.terminate() m.terminate()
l.terminate() l.terminate()
return reader() return reader
def train(): def train():
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册