提交 a8868a17 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Make the intent clearer in the final GC pass.

上级 78205812
...@@ -408,7 +408,7 @@ class Stack(VM): ...@@ -408,7 +408,7 @@ class Stack(VM):
# lazy evaluation. See discussion on theano-dev June 19 2012. # lazy evaluation. See discussion on theano-dev June 19 2012.
if self.allow_gc: if self.allow_gc:
for v in storage_map: for v in storage_map:
if v.owner and 'output' not in zip(*v.clients)[0]: if v.owner and not v in self.outputs:
storage_map[v][0] = None storage_map[v][0] = None
compute_map[v][0] = 0 compute_map[v][0] = 0
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论