diff --git a/demo/mnist/api_train_v2.py b/demo/mnist/api_train_v2.py index 67e36e6b201f3858edb0a4047b45a9e60fb3a9a5..d5eeb05318830050b27da9931d367880eb1c607e 100644 --- a/demo/mnist/api_train_v2.py +++ b/demo/mnist/api_train_v2.py @@ -36,11 +36,6 @@ def main(): trainer = paddle.trainer.SGD(update_equation=adam_optimizer) - reader = paddle.reader.batched( - paddle.reader.shuffle( - paddle.dataset.mnist.train_creator(), buf_size=8192), - batch_size=32) - trainer.train( train_reader=paddle.reader.batched( paddle.reader.shuffle(paddle.dataset.mnist.train_creator(),