From e8eb81ca282b0e56c2cdc085a3debcb5fab6c7ab Mon Sep 17 00:00:00 2001 From: yuyang18 Date: Fri, 13 Jul 2018 16:04:08 +0800 Subject: [PATCH] Just train two batch --- python/paddle/fluid/layers/io.py | 2 +- python/paddle/fluid/tests/demo/pyreader.py | 4 +++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/python/paddle/fluid/layers/io.py b/python/paddle/fluid/layers/io.py index 03788b41cb..149b33334a 100644 --- a/python/paddle/fluid/layers/io.py +++ b/python/paddle/fluid/layers/io.py @@ -520,7 +520,7 @@ def py_reader(capacity, shapes, dtypes, lod_levels=None, name=None): startup_var = startup_blk.create_var(name=reader_name) startup_blk.append_op( type='create_py_reader', - inputs={'blocking_queue': queue_name}, + inputs={'blocking_queue': [queue_name]}, outputs={'Out': [startup_var]}, attrs={ 'shape_concat': shape_concat, diff --git a/python/paddle/fluid/tests/demo/pyreader.py b/python/paddle/fluid/tests/demo/pyreader.py index 9fddef3c24..cc459a8f76 100644 --- a/python/paddle/fluid/tests/demo/pyreader.py +++ b/python/paddle/fluid/tests/demo/pyreader.py @@ -15,6 +15,7 @@ import paddle.fluid as fluid import paddle.dataset.mnist as mnist import paddle +import paddle.v2 import threading import numpy @@ -91,7 +92,8 @@ def main(): for epoch_id in xrange(10): train_data_thread = pipe_reader_to_queue( - paddle.batch(mnist.train(), 32), train_queue) + paddle.batch(paddle.v2.reader.firstn(mnist.train(), 32), 64), + train_queue) try: while True: print 'train_loss', numpy.array( -- GitLab