From 2ceeb2e75a5029ff7ff9dbee76f854a98c81c8ad Mon Sep 17 00:00:00 2001 From: dzhwinter Date: Tue, 5 Dec 2017 06:43:31 -0800 Subject: [PATCH] fix dataset --- paddle/understand_sentiment_lstm.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/paddle/understand_sentiment_lstm.py b/paddle/understand_sentiment_lstm.py index 79b3647..e1ebd03 100644 --- 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)