提交 9706d06e authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Use change_flags consitently.

上级 8af39e9c
...@@ -2228,17 +2228,11 @@ class _Maker(FunctionMaker): # inheritance buys a few helper functions ...@@ -2228,17 +2228,11 @@ class _Maker(FunctionMaker): # inheritance buys a few helper functions
inputs, outputs, accept_inplace) inputs, outputs, accept_inplace)
fgraph.equivalence_tracker = equivalence_tracker fgraph.equivalence_tracker = equivalence_tracker
# optimize the fgraph with change_flags(compute_test_value=config.compute_test_value_opt):
compute_test_value_orig = theano.config.compute_test_value
try:
theano.config.compute_test_value = \
theano.config.compute_test_value_opt
optimizer(fgraph) optimizer(fgraph)
theano.compile.function_module.insert_deepcopy( theano.compile.function_module.insert_deepcopy(
fgraph, inputs, list(chain(outputs, additional_outputs))) fgraph, inputs, list(chain(outputs, additional_outputs)))
finally:
theano.config.compute_test_value = compute_test_value_orig
if i == 0: if i == 0:
fgraph0 = fgraph fgraph0 = fgraph
...@@ -2287,7 +2281,7 @@ class _Maker(FunctionMaker): # inheritance buys a few helper functions ...@@ -2287,7 +2281,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'): with change_flags(compute_test_value=config.compute_test_value_opt):
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 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论