From 57a93acc86d5e0c8b9d0ac81770ecce4dab657b7 Mon Sep 17 00:00:00 2001 From: wangmeng28 Date: Sat, 20 Jan 2018 14:28:04 +0800 Subject: [PATCH] Fix typo in seresnext --- fluid/image_classification/se_resnext.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/fluid/image_classification/se_resnext.py b/fluid/image_classification/se_resnext.py index 46b938f1..886cbc5f 100644 --- a/fluid/image_classification/se_resnext.py +++ b/fluid/image_classification/se_resnext.py @@ -125,8 +125,8 @@ def train(learning_rate, batch_size, num_passes, model_save_dir='model'): exe = fluid.Executor(place) exe.run(fluid.default_startup_program()) - train_reader = paddle.batch(datareader.train(), batch_size=batch_size) - test_reader = paddle.batch(datareader.test(), batch_size=batch_size) + train_reader = paddle.batch(reader.train(), batch_size=batch_size) + test_reader = paddle.batch(reader.test(), batch_size=batch_size) feeder = fluid.DataFeeder(place=place, feed_list=[image, label]) for pass_id in range(num_passes): -- GitLab