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

remove optimiser param from __init__.py

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