From 9fe65642edd10e4b3a9d1266eb5b4d0b3637aa07 Mon Sep 17 00:00:00 2001 From: zhengya01 Date: Wed, 30 Jan 2019 10:42:25 +0000 Subject: [PATCH] update metric_learning --- fluid/PaddleCV/metric_learning/train_elem.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/fluid/PaddleCV/metric_learning/train_elem.py b/fluid/PaddleCV/metric_learning/train_elem.py index c3c0a52b..8de4b667 100644 --- a/fluid/PaddleCV/metric_learning/train_elem.py +++ b/fluid/PaddleCV/metric_learning/train_elem.py @@ -233,13 +233,9 @@ def train_async(args): totalruntime += period total_time += 1 - #if iter_no % args.test_iter_step == 0 and iter_no != 0: - if (iter_no % args.test_iter_step == 0 and iter_no != 0) or args.enable_ce: + if iter_no % args.test_iter_step == 0 and iter_no != 0: f, l = [], [] for batch_id, data in enumerate(test_reader()): - if args.enable_ce: - if batch_id > 1: - break t1 = time.time() [feas] = exe.run(test_prog, fetch_list = test_fetch_list, feed=test_feeder.feed(data)) -- GitLab