From bd80549c6525f2f4c20ecab7a2305155c2c06462 Mon Sep 17 00:00:00 2001 From: Helin Wang Date: Mon, 20 Feb 2017 09:03:55 -0800 Subject: [PATCH] fix issue in create custom reader in design doc --- doc/design/reader/README.md | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/doc/design/reader/README.md b/doc/design/reader/README.md index 8f7abf12f73..ff3ee758bee 100644 --- a/doc/design/reader/README.md +++ b/doc/design/reader/README.md @@ -107,7 +107,7 @@ We decided to use dictionary (`{"image":0, "label":1}`) instead of list (`["imag ### How to create custom data reader ```python -def image_reader(image_path, label_path): +def image_reader(image_path, label_path, n): f = open(image_path) l = open(label_path) images = numpy.fromfile( @@ -117,9 +117,10 @@ def image_reader(image_path, label_path): for i in xrange(n): yield images[i, :], labels[i] # a single entry of data is created each time f.close() + l.close() # use python lambda to change image_reader into a function with no parameters. -reader = lambda : image_reader("/path/to/image_file", "/path/to/label_file") +reader = lambda : image_reader("/path/to/image_file", "/path/to/label_file", 1024) paddle.train(reader, {"image":0, "label":1}, ...) ``` -- GitLab