提交 09a5b97e authored 作者: affanv14's avatar affanv14

remove optimiser param from __init__.py

上级 1fe35b04
......@@ -1131,12 +1131,12 @@ class LocalMetaOptimizer(LocalOptimizer):
"""
def __init__(self, optimizers=()):
self._tracks = [x for o in optimizers for x in o.tracks()]
self.optimizers = list(optimizers)
def __init__(self):
self.verbose = config.metaopt.verbose
self.track_dict = defaultdict(lambda: [])
self.tag_dict = defaultdict(lambda: [])
self._tracks = []
self.optimizers = []
def register(self, optimizer, tag_list):
self.optimizers.append(optimizer)
......
......@@ -2007,8 +2007,8 @@ def local_abstractconv3d_gradinputs_gemm_alt(node):
class ConvMetaOptimizer(LocalMetaOptimizer):
def __init__(self, optimizers=()):
super(ConvMetaOptimizer, self).__init__(optimizers)
def __init__(self):
super(ConvMetaOptimizer, self).__init__()
def time_call(self, fn):
start = time.time()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论