From 6e38de1ce2c096a33509279252e413dc8f6d41e4 Mon Sep 17 00:00:00 2001 From: yuyang18 Date: Fri, 13 Jul 2018 15:15:06 +0800 Subject: [PATCH] Strange Hang problem --- python/paddle/fluid/tests/demo/pyreader.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/python/paddle/fluid/tests/demo/pyreader.py b/python/paddle/fluid/tests/demo/pyreader.py index 47b20a0c5c..356a27cdd4 100644 --- a/python/paddle/fluid/tests/demo/pyreader.py +++ b/python/paddle/fluid/tests/demo/pyreader.py @@ -100,10 +100,11 @@ def main(): train_data_thread.join() test_data_thread = pipe_reader_to_queue( - paddle.batch(mnist.train(), 32), test_queue) + paddle.batch(mnist.test(), 32), test_queue) try: while True: - print numpy.array(tester.run(fetch_list=[test_loss.name])) + print 'test loss', numpy.array( + tester.run(fetch_list=[test_loss.name])) except fluid.core.EOFException: print 'End of testing' test_reader.reset() -- GitLab