提交 02b1a99e 编写于 作者: S sunyanfang01

modify log in prune

上级 5a8c902c
...@@ -42,7 +42,7 @@ def sensitivity(program, ...@@ -42,7 +42,7 @@ def sensitivity(program,
if pruned_ratios is None: if pruned_ratios is None:
pruned_ratios = np.arange(0.1, 1, step=0.1) pruned_ratios = np.arange(0.1, 1, step=0.1)
total_evaluate_iters = 1 total_evaluate_iters = 0
for name in param_names: for name in param_names:
if name not in sensitivities: if name not in sensitivities:
sensitivities[name] = {} sensitivities[name] = {}
...@@ -52,12 +52,6 @@ def sensitivity(program, ...@@ -52,12 +52,6 @@ def sensitivity(program,
len(list(pruned_ratios)) - len(sensitivities[name])) len(list(pruned_ratios)) - len(sensitivities[name]))
eta = '-' eta = '-'
start_time = time.time() start_time = time.time()
progress = 1.0 / total_evaluate_iters
progress = "%.2f%%" % (progress * 100)
logging.info(
"Total evaluate iters={}, current={}, progress={}, eta={}".format(
total_evaluate_iters, 1, progress, eta),
use_color=True)
baseline = eval_func(graph.program) baseline = eval_func(graph.program)
cost = time.time() - start_time cost = time.time() - start_time
eta = cost * (total_evaluate_iters - 1) eta = cost * (total_evaluate_iters - 1)
...@@ -73,7 +67,7 @@ def sensitivity(program, ...@@ -73,7 +67,7 @@ def sensitivity(program,
logging.info( logging.info(
"Total evaluate iters={}, current={}, progress={}, eta={}". "Total evaluate iters={}, current={}, progress={}, eta={}".
format( format(
total_evaluate_iters, current_iter+1, progress, total_evaluate_iters, current_iter, progress,
seconds_to_hms( seconds_to_hms(
int(cost * (total_evaluate_iters - current_iter)))), int(cost * (total_evaluate_iters - current_iter)))),
use_color=True) use_color=True)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册