diff --git a/fluid/image_classification/train.py b/fluid/image_classification/train.py index 6244e520900b58914b847c4acb451beb252efd30..4f1264281852094b2e0346ff9564ebb5986edf73 100644 --- a/fluid/image_classification/train.py +++ b/fluid/image_classification/train.py @@ -220,8 +220,7 @@ def train_parallel_exe(args, test_info = [[], [], []] for batch_id, data in enumerate(train_reader()): t1 = time.time() - loss, acc1, acc5 = train_exe.run(fetch_list, - feed_dict=feeder.feed(data)) + loss, acc1, acc5 = train_exe.run(fetch_list, feed=feeder.feed(data)) t2 = time.time() period = t2 - t1 loss = np.mean(np.array(loss)) @@ -243,8 +242,7 @@ def train_parallel_exe(args, train_acc5 = np.array(train_info[2]).mean() for data in test_reader(): t1 = time.time() - loss, acc1, acc5 = test_exe.run(fetch_list, - feed_dict=feeder.feed(data)) + loss, acc1, acc5 = test_exe.run(fetch_list, feed=feeder.feed(data)) t2 = time.time() period = t2 - t1 loss = np.mean(np.array(loss))