提交 e5fc248c authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Make sure to disable compute_test_values for OutputGuard in DebugMode.

上级 048d0c47
...@@ -28,6 +28,7 @@ from theano.compile.function_module import ( ...@@ -28,6 +28,7 @@ from theano.compile.function_module import (
std_fgraph) std_fgraph)
from theano.compile.mode import Mode, register_mode from theano.compile.mode import Mode, register_mode
from theano.compile.ops import OutputGuard, _output_guard from theano.compile.ops import OutputGuard, _output_guard
from theano.configparser import change_flags
__docformat__ = "restructuredtext en" __docformat__ = "restructuredtext en"
...@@ -2286,6 +2287,7 @@ class _Maker(FunctionMaker): # inheritance buys a few helper functions ...@@ -2286,6 +2287,7 @@ class _Maker(FunctionMaker): # inheritance buys a few helper functions
fgraph.attach_feature(gof.DestroyHandler()) fgraph.attach_feature(gof.DestroyHandler())
for o in fgraph.outputs: for o in fgraph.outputs:
try: try:
with change_flags(compute_test_value='off'):
fgraph.replace_validate(o, _output_guard(o), reason='output_guard') fgraph.replace_validate(o, _output_guard(o), reason='output_guard')
raise Exception("Output variable %s required output_guard, " raise Exception("Output variable %s required output_guard, "
"how was this output left unprotected against " "how was this output left unprotected against "
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论