From 1524cd88e515cd0438a7d26a5ba1b9cc713b5785 Mon Sep 17 00:00:00 2001 From: Yibing Liu Date: Wed, 25 Apr 2018 21:16:16 -0700 Subject: [PATCH] Remove test code --- fluid/image_classification/train.py | 9 --------- 1 file changed, 9 deletions(-) diff --git a/fluid/image_classification/train.py b/fluid/image_classification/train.py index bc7af517..f0cc8b1d 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) -- GitLab