提交 c9990cc4 authored 作者: Frederic's avatar Frederic

Make the MonitorMode use the default optimizer.

上级 7052c553
...@@ -21,7 +21,7 @@ class MonitorMode(Mode): ...@@ -21,7 +21,7 @@ class MonitorMode(Mode):
""" """
def __init__(self, pre_func=None, post_func=None, def __init__(self, pre_func=None, post_func=None,
optimizer='fast_run', linker=None): optimizer='default', linker=None):
""" """
Constructor. Constructor.
...@@ -44,10 +44,13 @@ class MonitorMode(Mode): ...@@ -44,10 +44,13 @@ class MonitorMode(Mode):
self.post_func = post_func self.post_func = post_func
wrap_linker = theano.gof.WrapLinkerMany([theano.gof.OpWiseCLinker()], wrap_linker = theano.gof.WrapLinkerMany([theano.gof.OpWiseCLinker()],
[self.eval]) [self.eval])
if optimizer is 'default':
optimizer = theano.config.optimizer
if (linker is not None and if (linker is not None and
not isinstance(linker.mode, MonitorMode)): not isinstance(linker.mode, MonitorMode)):
raise Exception("MonitorMode can only use its own linker! You " raise Exception("MonitorMode can only use its own linker! You "
"should not provide one.", linker) "should not provide one.", linker)
super(MonitorMode, self).__init__(wrap_linker, optimizer=optimizer) super(MonitorMode, self).__init__(wrap_linker, optimizer=optimizer)
def eval(self, i, node, fn): def eval(self, i, node, fn):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论