提交 405397d0 authored 作者: Roy Xue's avatar Roy Xue

if condition fix

上级 a5dc7a9b
...@@ -749,8 +749,8 @@ class ProfileStats(object): ...@@ -749,8 +749,8 @@ class ProfileStats(object):
origin = view_of[ins] origin = view_of[ins]
viewed_by[origin].remove(ins) viewed_by[origin].remove(ins)
if (not viewed_by[origin] and if (not viewed_by[origin] and
origin not in fgraph.inputs): origin not in fgraph.inputs and
not isinstance(origin, theano.Constant)):
running_memory_size -= var_mem[origin] running_memory_size -= var_mem[origin]
else: else:
# ins is viewed_by something else, so its # ins is viewed_by something else, so its
...@@ -878,7 +878,9 @@ class ProfileStats(object): ...@@ -878,7 +878,9 @@ class ProfileStats(object):
elif ins in view_of_temp: elif ins in view_of_temp:
origin = view_of_temp[ins] origin = view_of_temp[ins]
viewed_by_temp[origin].remove(ins) viewed_by_temp[origin].remove(ins)
if not viewed_by_temp[origin] and origin not in fgraph.inputs: if (not viewed_by_temp[origin] and
origin not in fgraph.inputs and
not isinstance(origin, theano.Constant)):
mem_freed += var_mem[origin] mem_freed += var_mem[origin]
else: else:
# ins is viewed_by something else, so its # ins is viewed_by something else, so its
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论