提交 7f249045 authored 作者: Frederic's avatar Frederic

Better auto function name if we call var.eval().

上级 48c63a85
...@@ -195,8 +195,12 @@ def function(inputs, outputs=None, mode=None, updates=None, givens=None, ...@@ -195,8 +195,12 @@ def function(inputs, outputs=None, mode=None, updates=None, givens=None,
last_frame = stack[idx] last_frame = stack[idx]
if (last_frame[0] == source_file or last_frame[0] == compiled_file): if (last_frame[0] == source_file or last_frame[0] == compiled_file):
func_frame = stack[idx - 1] func_frame = stack[idx - 1]
while "theano/gof" in func_frame[0] and idx > 0:
idx -= 1
# This can hapen if we call var.eval()
func_frame = stack[idx - 1]
name = func_frame[0] + ':' + str(func_frame[1]) name = func_frame[0] + ':' + str(func_frame[1])
if updates is None: if updates is None:
updates = [] updates = []
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论