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

Don't execute compute_test_value during optimization.

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