From 10ee0661c7cb03346116f4b9fd1f95705eea5708 Mon Sep 17 00:00:00 2001
From: wanghaoshuang <wanghaoshuang@baidu.com>
Date: Thu, 16 Nov 2017 17:29:40 +0800
Subject: [PATCH] fix data reader error

---
 text_classification/train.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/text_classification/train.py b/text_classification/train.py
index cda04bfc..888fde35 100644
--- a/text_classification/train.py
+++ b/text_classification/train.py
@@ -46,10 +46,10 @@ def train(topology,
         word_dict = paddle.dataset.imdb.word_dict()
         train_reader = paddle.batch(
             paddle.reader.shuffle(
-                lambda: paddle.dataset.imdb.train(word_dict), buf_size=1000),
+                lambda: paddle.dataset.imdb.train(word_dict)(), buf_size=1000),
             batch_size=100)
         test_reader = paddle.batch(
-            lambda: paddle.dataset.imdb.test(word_dict), batch_size=100)
+            lambda: paddle.dataset.imdb.test(word_dict)(), batch_size=100)
 
         class_num = 2
     else:
-- 
GitLab