提交 0729d792 authored 作者: affanv14's avatar affanv14

make metaoptimizer optimise faster

上级 ce2d8f86
......@@ -1191,7 +1191,8 @@ class LocalMetaOptimizer(LocalOptimizer):
try:
fn = theano.function([], outputs, givens=givens,
on_unused_input='ignore')
timing = min(self.time_call(fn) for _ in range(3))
fn.trust_input = True
timing = min(self.time_call(fn) for _ in range(2))
except Exception as e:
if self.verbose > 0:
print("* %s: exception" % opt, e)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论