提交 58a0f97b authored 作者: Frederic's avatar Frederic

Remove the change_tracker feature when we stop to use it.

上级 f6e80455
...@@ -1742,13 +1742,14 @@ class EquilibriumOptimizer(NavigatorOptimizer): ...@@ -1742,13 +1742,14 @@ class EquilibriumOptimizer(NavigatorOptimizer):
def add_requirements(self, fgraph): def add_requirements(self, fgraph):
super(EquilibriumOptimizer, self).add_requirements(fgraph) super(EquilibriumOptimizer, self).add_requirements(fgraph)
fgraph.attach_feature(ChangeTracker())
for opt in self.get_local_optimizers(): for opt in self.get_local_optimizers():
opt.add_requirements(fgraph) opt.add_requirements(fgraph)
for opt in self.global_optimizers: for opt in self.global_optimizers:
opt.add_requirements(fgraph) opt.add_requirements(fgraph)
def apply(self, fgraph, start_from=None): def apply(self, fgraph, start_from=None):
change_tracker = ChangeTracker()
fgraph.attach_feature(change_tracker)
if start_from is None: if start_from is None:
start_from = fgraph.outputs start_from = fgraph.outputs
else: else:
...@@ -1853,7 +1854,7 @@ class EquilibriumOptimizer(NavigatorOptimizer): ...@@ -1853,7 +1854,7 @@ class EquilibriumOptimizer(NavigatorOptimizer):
+ ". You can safely raise the current threshold of " + ". You can safely raise the current threshold of "
+ "%f with the theano flag 'optdb.max_use_ratio'." % + "%f with the theano flag 'optdb.max_use_ratio'." %
config.optdb.max_use_ratio) config.optdb.max_use_ratio)
fgraph.remove_feature(change_tracker)
return (self, loop_timing, loop_process_count, return (self, loop_timing, loop_process_count,
(start_nb_nodes, end_nb_nodes, max_nb_nodes), (start_nb_nodes, end_nb_nodes, max_nb_nodes),
global_opt_timing, nb_nodes, time_opts, io_toposort_timing) global_opt_timing, nb_nodes, time_opts, io_toposort_timing)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论