提交 e629abd5 authored 作者: James Bergstra's avatar James Bergstra

added an assert in EquilibriumOptimizer

上级 f14f5df1
...@@ -875,6 +875,8 @@ class EquilibriumOptimizer(NavigatorOptimizer): ...@@ -875,6 +875,8 @@ class EquilibriumOptimizer(NavigatorOptimizer):
while changed and not max_use_abort: while changed and not max_use_abort:
changed = False changed = False
for node in start_from:
assert node in env.outputs
q = deque(graph.io_toposort(env.inputs, start_from)) q = deque(graph.io_toposort(env.inputs, start_from))
...@@ -914,6 +916,7 @@ class EquilibriumOptimizer(NavigatorOptimizer): ...@@ -914,6 +916,7 @@ class EquilibriumOptimizer(NavigatorOptimizer):
def _check_chain(r, chain): def _check_chain(r, chain):
"""WRITEME""" """WRITEME"""
chain = list(reversed(chain)) chain = list(reversed(chain))
while chain: while chain:
elem = chain.pop() elem = chain.pop()
...@@ -933,7 +936,10 @@ def _check_chain(r, chain): ...@@ -933,7 +936,10 @@ def _check_chain(r, chain):
return False return False
if chain: if chain:
r = r.owner.inputs[chain.pop()] r = r.owner.inputs[chain.pop()]
#print 'check_chain', _check_chain.n_calls
#_check_chain.n_calls += 1
return r return r
#_check_chain.n_calls = 0
def check_chain(r, *chain): def check_chain(r, *chain):
"""WRITEME""" """WRITEME"""
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论