fix bug (#5233)

7 years ago · a75437a20c
parent 4fdbad010e
commit a75437a20c
1 changed files with 1 additions and 1 deletions
--- a/python/paddle/v2/dataset/imdb.py
+++ b/python/paddle/v2/dataset/imdb.py
@ -116,7 +116,7 @@ def reader_creator(pos_pattern, neg_pattern, word_idx, buffer_size):
            yield [word_idx.get(w, UNK) for w in doc], i % 2
            doc = qs[i % 2].get()

-    return reader()
+    return reader


 def train(word_idx):