提交 cb31330d authored 作者: Frederic Bastien's avatar Frederic Bastien

Don't execute compute_test_value during optimization.

上级 445264c7
......@@ -937,9 +937,14 @@ class FunctionMaker(object):
optimizer, linker = mode.optimizer, copy.copy(mode.linker)
# optimize the env
start_optimizer = time.time()
optimizer(env)
end_optimizer = time.time()
compute_test_value_orig = theano.config.compute_test_value
try:
theano.config.compute_test_value = "ignore"
start_optimizer = time.time()
optimizer(env)
end_optimizer = time.time()
finally:
theano.config.compute_test_value = compute_test_value_orig
mode.optimizer_time += end_optimizer - start_optimizer
_logger.debug('Optimizing took %f seconds' % (end_optimizer - start_optimizer))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论