提交 6649a6ab authored 作者: Frederic Bastien's avatar Frederic Bastien

Disable compute_test_value during Theano optimization phase. This give a better…

Disable compute_test_value during Theano optimization phase. This give a better error in pymc3 issue 1939
上级 bc897190
...@@ -1475,7 +1475,8 @@ class FunctionMaker(object): ...@@ -1475,7 +1475,8 @@ 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:
optimizer_profile = optimizer(fgraph) with theano.configparser.change_flags(compute_test_value="off"):
optimizer_profile = optimizer(fgraph)
end_optimizer = time.time() end_optimizer = time.time()
opt_time = end_optimizer - start_optimizer opt_time = end_optimizer - start_optimizer
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论