Merge pull request #8 from dzhwinter/fix/dataset

fix dataset
dzhwinter · Dec 5, 2017 · 9529663 · 9529663
2 parents 81461bc + 2ceeb2e
commit 9529663
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/paddle/understand_sentiment_lstm.py b/paddle/understand_sentiment_lstm.py
@@ -139,7 +139,8 @@ def run_benchmark(model, args):
 
     train_reader = paddle.batch(
         paddle.reader.shuffle(
-            paddle.dataset.flowers.train(), buf_size=args.batch_size * 10),
+            paddle.dataset.imdb.train(word_dict),
+            buf_size=25000),  # only set imdb for speed
         batch_size=args.batch_size)
     place = fluid.CPUPlace() if args.device == 'CPU' else fluid.GPUPlace(0)
     exe = fluid.Executor(place)