提交 9fe65642 编写于 作者: Z zhengya01

update metric_learning

上级 4144c6a6
...@@ -233,13 +233,9 @@ def train_async(args): ...@@ -233,13 +233,9 @@ def train_async(args):
totalruntime += period totalruntime += period
total_time += 1 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:
if (iter_no % args.test_iter_step == 0 and iter_no != 0) or args.enable_ce:
f, l = [], [] f, l = [], []
for batch_id, data in enumerate(test_reader()): for batch_id, data in enumerate(test_reader()):
if args.enable_ce:
if batch_id > 1:
break
t1 = time.time() t1 = time.time()
[feas] = exe.run(test_prog, fetch_list = test_fetch_list, feed=test_feeder.feed(data)) [feas] = exe.run(test_prog, fetch_list = test_fetch_list, feed=test_feeder.feed(data))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册