提交 d2309acb authored 作者: Frederic Bastien's avatar Frederic Bastien

print the nb nodes before/after the opt.

上级 fe78d0c2
...@@ -315,17 +315,17 @@ class SeqOptimizer(Optimizer, list): ...@@ -315,17 +315,17 @@ class SeqOptimizer(Optimizer, list):
" time - (name, class, index, nodes before, nodes after) - validate time", " time - (name, class, index, nodes before, nodes after) - validate time",
file=stream) file=stream)
ll = [] ll = []
for opt in opts: for (opt, nb_n) in zip(opts, nb_nodes):
if hasattr(opt, "__name__"): if hasattr(opt, "__name__"):
name = opt.__name__ name = opt.__name__
else: else:
name = opt.name name = opt.name
idx = opts.index(opt) idx = opts.index(opt)
ll.append((name, opt.__class__.__name__, ll.append((name, opt.__class__.__name__,
idx)) idx) + nb_n)
lll = sorted(zip(prof, ll, nb_nodes), key=lambda a: a[0]) lll = sorted(zip(prof, ll), key=lambda a: a[0])
for (t, opt, nb_n) in lll[::-1]: for (t, opt) in lll[::-1]:
i = opt[2] i = opt[2]
if sub_validate_time: if sub_validate_time:
val_time = sub_validate_time[i + 1] - sub_validate_time[i] val_time = sub_validate_time[i + 1] - sub_validate_time[i]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论