提交 6798e88c authored 作者: f0k's avatar f0k

Rename config flag conv_meta.verbose to metaopt.verbose

上级 1fdf3bd3
...@@ -823,6 +823,10 @@ class LocalOptimizer(object): ...@@ -823,6 +823,10 @@ class LocalOptimizer(object):
(' ' * level), self.__class__.__name__, id(self)) (' ' * level), self.__class__.__name__, id(self))
theano.configparser.AddConfigVar('metaopt.verbose',
"Enable verbose output for meta optimizers",
theano.configparser.BoolParam(False), in_c_key=False)
class LocalMetaOptimizer(LocalOptimizer): class LocalMetaOptimizer(LocalOptimizer):
"""Base class for meta-optimizers that try a set of LocalOptimizers """Base class for meta-optimizers that try a set of LocalOptimizers
to replace a node and choose the one that executes the fastest""" to replace a node and choose the one that executes the fastest"""
...@@ -830,7 +834,7 @@ class LocalMetaOptimizer(LocalOptimizer): ...@@ -830,7 +834,7 @@ class LocalMetaOptimizer(LocalOptimizer):
def __init__(self, tracks=None, optimizers=()): def __init__(self, tracks=None, optimizers=()):
self._tracks = tracks self._tracks = tracks
self.optimizers = list(optimizers) self.optimizers = list(optimizers)
self.verbose = False self.verbose = config.metaopt.verbose
def register(self, optimizer): def register(self, optimizer):
self.optimizers.append(optimizer) self.optimizers.append(optimizer)
......
...@@ -1361,14 +1361,9 @@ class LocalCudaMetaOptimizer(LocalMetaOptimizer): ...@@ -1361,14 +1361,9 @@ class LocalCudaMetaOptimizer(LocalMetaOptimizer):
# Convolution Meta-optimizer # Convolution Meta-optimizer
theano.configparser.AddConfigVar('conv_meta.verbose',
"Enable verbose output for conv_meta optimizer",
theano.configparser.BoolParam(False), in_c_key=False)
class ConvMetaOptimizer(LocalCudaMetaOptimizer): class ConvMetaOptimizer(LocalCudaMetaOptimizer):
def __init__(self, optimizers): def __init__(self, optimizers):
super(ConvMetaOptimizer, self).__init__([GpuConv], optimizers) super(ConvMetaOptimizer, self).__init__([GpuConv], optimizers)
self.verbose = config.conv_meta.verbose
def provide_inputs(self, node, inputs): def provide_inputs(self, node, inputs):
# We need to provide dummy data for the given inputs. # We need to provide dummy data for the given inputs.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论