提交 50dc6640 authored 作者: Ziye Fan's avatar Ziye Fan

fix final_opt profiling time calc

上级 c9187dec
......@@ -1852,6 +1852,7 @@ class EquilibriumOptimizer(NavigatorOptimizer):
self.detach_updater(fgraph, u)
# Apply final optimizers
t_before_final_opt = time.time()
for gopt in self.final_optimizers:
change_tracker.reset()
nb = change_tracker.nb_imported
......@@ -1869,7 +1870,7 @@ class EquilibriumOptimizer(NavigatorOptimizer):
opt_name = (getattr(gopt, "name", None)
or getattr(gopt, "__name__", ""))
global_opt_timing.append(float(time.time() - t0))
global_opt_timing[-1] += time.time() - t_before_final_opt
loop_process_count.append(process_count)
loop_timing.append(float(time.time() - t0))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论