提交 1fdd4c84 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Disable compute_test_value in DebugMode's optimizer too.

上级 164b6ebc
......@@ -1414,8 +1414,14 @@ class _Maker(FunctionMaker): #inheritance buys a few helper functions
for i in xrange(mode.stability_patience):
env, additional_outputs, equivalence_tracker = _optcheck_env(expanded_inputs, outputs, accept_inplace)
env.equivalence_tracker = equivalence_tracker
# optimize the env
optimizer(env)
compute_test_value_orig = theano.config.compute_test_value
try:
theano.config.compute_test_value = "off"
optimizer(env)
finally:
theano.config.compute_test_value = compute_test_value_orig
theano.compile.function_module.insert_deepcopy(env, inputs, outputs+additional_outputs)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论