提交 38d38f11 authored 作者: Frederic Bastien's avatar Frederic Bastien

better error msg and del a var to help find a memory leak.

上级 8604fcc2
...@@ -131,7 +131,7 @@ def pfunc(params, outputs=None, mode=None, updates=[], givens=[]): ...@@ -131,7 +131,7 @@ def pfunc(params, outputs=None, mode=None, updates=[], givens=[]):
cloned_outputs = list(computed_list) cloned_outputs = list(computed_list)
else: else:
if not isinstance(outputs, Variable): if not isinstance(outputs, Variable):
raise TypeError('output must be a theano Variable instance', outputs) raise TypeError('outputs must be a theano Variable instance or list of.', outputs)
cloned_outputs = v_clone(outputs) cloned_outputs = v_clone(outputs)
computed_list = [cloned_outputs] computed_list = [cloned_outputs]
...@@ -163,6 +163,7 @@ def pfunc(params, outputs=None, mode=None, updates=[], givens=[]): ...@@ -163,6 +163,7 @@ def pfunc(params, outputs=None, mode=None, updates=[], givens=[]):
inputs += [In(variable=si, value=si.container, mutable=False) inputs += [In(variable=si, value=si.container, mutable=False)
for si in shared_inputs for si in shared_inputs
if si not in set_of_param_variables] if si not in set_of_param_variables]
del shared_inputs
# Iterate over the updates, which are either pairs # Iterate over the updates, which are either pairs
# (shared_var, expressionvariable), or a similar dictionary. # (shared_var, expressionvariable), or a similar dictionary.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论