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

Fix reader creator in mnist

上级 e55ec44b
......@@ -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,
......
......@@ -45,7 +45,7 @@ def reader_creator(image_filename, label_filename, buffer_size):
m.terminate()
l.terminate()
return reader()
return reader
def train():
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册