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

small crash fix

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