Strange Hang problem

7 years ago · 6e38de1ce2
parent 8c2eba71e2
commit 6e38de1ce2
1 changed files with 3 additions and 2 deletions
--- 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()