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

disable test value during the optimization AND the creation of thunks.

上级 6649a6ab
...@@ -1475,8 +1475,7 @@ class FunctionMaker(object): ...@@ -1475,8 +1475,7 @@ class FunctionMaker(object):
optimizer_profile = self.optimize_graph_with_cache( optimizer_profile = self.optimize_graph_with_cache(
optimizer, inputs, outputs) optimizer, inputs, outputs)
else: else:
with theano.configparser.change_flags(compute_test_value="off"): optimizer_profile = optimizer(fgraph)
optimizer_profile = optimizer(fgraph)
end_optimizer = time.time() end_optimizer = time.time()
opt_time = end_optimizer - start_optimizer opt_time = end_optimizer - start_optimizer
...@@ -1800,14 +1799,15 @@ def orig_function(inputs, outputs, mode=None, accept_inplace=False, ...@@ -1800,14 +1799,15 @@ def orig_function(inputs, outputs, mode=None, accept_inplace=False,
fn = None fn = None
try: try:
Maker = getattr(mode, 'function_maker', FunctionMaker) Maker = getattr(mode, 'function_maker', FunctionMaker)
fn = Maker(inputs, with theano.configparser.change_flags(compute_test_value="off"):
outputs, fn = Maker(inputs,
mode, outputs,
accept_inplace=accept_inplace, mode,
profile=profile, accept_inplace=accept_inplace,
on_unused_input=on_unused_input, profile=profile,
output_keys=output_keys).create( on_unused_input=on_unused_input,
defaults) output_keys=output_keys).create(
defaults)
finally: finally:
t2 = time.time() t2 = time.time()
if fn and profile: if fn and profile:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论