From 9538969d54fe2f46d3e72338ddd1a8e27f252db3 Mon Sep 17 00:00:00 2001 From: ruri Date: Fri, 13 Mar 2020 16:38:19 +0800 Subject: [PATCH] cherry_pick mannual (#4422) --- PaddleCV/image_classification/eval.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/PaddleCV/image_classification/eval.py b/PaddleCV/image_classification/eval.py index 44ea4736..9653af80 100644 --- a/PaddleCV/image_classification/eval.py +++ b/PaddleCV/image_classification/eval.py @@ -113,7 +113,7 @@ def eval(args): test_program = fluid.default_main_program().clone(for_test=True) - fetch_list = [avg_cost.name, acc_top1.name, acc_top5.name, pred.name] + fetch_list = [avg_cost.name, acc_top1.name, acc_top5.name] gpu_id = int(os.environ.get('FLAGS_selected_gpus', 0)) place = fluid.CUDAPlace(gpu_id) if args.use_gpu else fluid.CPUPlace() @@ -151,7 +151,7 @@ def eval(args): parallel_data.append(image_data) if place_num == len(parallel_data): t1 = time.time() - loss_set, acc1_set, acc5_set, pred_set = exe.run( + loss_set, acc1_set, acc5_set = exe.run( compiled_program, fetch_list=fetch_list, feed=list(feeder.feed_parallel(parallel_data, place_num))) -- GitLab