diff --git a/fluid/PaddleCV/metric_learning/train_elem.py b/fluid/PaddleCV/metric_learning/train_elem.py index c3c0a52bce2c0d69ecc6d71d84634d48c8e82d87..8de4b667a98e0c4fa940e0d5ec493402e8ad6e86 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))