提交 29e67107 authored 作者: Frederic Bastien's avatar Frederic Bastien

Make sure fgraph attribute are always removed from apply node and variables when…

Make sure fgraph attribute are always removed from apply node and variables when they aren't in an fgraph anymore.
上级 b670565c
...@@ -334,7 +334,10 @@ class FunctionGraph(utils.object2): ...@@ -334,7 +334,10 @@ class FunctionGraph(utils.object2):
apply_node.tag.removed_by = [] apply_node.tag.removed_by = []
apply_node.tag.removed_by.append(str(reason)) apply_node.tag.removed_by.append(str(reason))
self.apply_nodes.remove(apply_node) self.apply_nodes.remove(apply_node)
del apply_node.fgraph
self.variables.difference_update(apply_node.outputs) self.variables.difference_update(apply_node.outputs)
for var in apply_node.outputs:
del var.fgraph
self.execute_callbacks('on_prune', apply_node, reason) self.execute_callbacks('on_prune', apply_node, reason)
for i, input in enumerate(apply_node.inputs): for i, input in enumerate(apply_node.inputs):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论