提交 3d3cbd94 authored 作者: Frederic's avatar Frederic

small crash fix

上级 7345f60d
......@@ -667,9 +667,9 @@ class ProfileStats(object):
v = out.type.get_size(sh)
sum_dense += v
else:
v = 'Unknown'
v = 0 # 'Unknown'
else:
v = 'Variable isnt created'
v = 0 # 'Variable isnt created'
var_mem[out] = v
fct_memory[node.fgraph][node].append(v)
......
......@@ -157,7 +157,7 @@ N.B.:
# If the op was compiled, print the optimized version.
outputs = s.owner.op.fn.maker.fgraph.outputs
else:
outputs = s.owner.op.output
outputs = s.owner.op.outputs
for idx, i in enumerate(outputs):
if hasattr(i, 'owner') and hasattr(i.owner, 'op'):
if isinstance(i.owner.op, theano.scan_module.scan_op.Scan):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论