提交 210d764a authored 作者: Roy Xue's avatar Roy Xue

fix some small details

上级 27406aae
......@@ -26,4 +26,6 @@ def test_profiling():
theano.config.profile = old1
theano.config.profile_memory = old2
test_profiling()
\ No newline at end of file
if __name__ = '__main__':
test_profiling()
\ No newline at end of file
......@@ -535,8 +535,12 @@ class Stack(VM):
if self.allow_gc:
for v in storage_map:
if v.owner and not v in self.outputs:
storage_map[v][0] = None
final_index.append(v)
if storage_map[v][0] = None or compute_map[v][0] = 2:
continue
else:
storage_map[v][0] = None
final_index.append(v)
compute_map[v][0] = 2
self.node_cleared_order.append(final_index)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论