diff --git a/fluid/image_classification/train.py b/fluid/image_classification/train.py index bc7af517cf41d8b96ffe7217d04e0e6cfc2f3ee7..f0cc8b1d71e5e15bd274c55b324149c59c08288a 100644 --- a/fluid/image_classification/train.py +++ b/fluid/image_classification/train.py @@ -227,15 +227,6 @@ def train_parallel_exe(args, train_reader = paddle.batch(reader.train(), batch_size=batch_size) test_reader = paddle.batch(reader.test(), batch_size=batch_size) - - # data reader - train_reader = paddle.batch( - paddle.reader.shuffle( - paddle.dataset.flowers.train(), buf_size=5120), - batch_size=batch_size) - test_reader = paddle.batch( - paddle.dataset.flowers.test(), batch_size=batch_size) - feeder = fluid.DataFeeder(place=place, feed_list=[image, label]) train_exe = fluid.ParallelExecutor(use_cuda=True, loss_name=avg_cost.name)