iicd/xautodl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add SuperSimpleNorm and update synthetic env

Browse Source
This commit is contained in:
D-X-Y
2021-04-23 02:12:11 -07:00
parent a5b7d986b3
commit 9b895bdf2e
13 changed files with 238 additions and 519 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
lib/xlayers/super_container.py
View File

@@ -91,6 +91,8 @@ class SuperSequential(SuperModule):
def abstract_search_space(self):
root_node = spaces.VirtualNode(id(self))
for index, module in enumerate(self):
if not isinstance(module, SuperModule):
continue
space = module.abstract_search_space
if not spaces.is_determined(space):
root_node.append(str(index), space)
@@ -98,9 +100,9 @@ class SuperSequential(SuperModule):
def apply_candidate(self, abstract_child: spaces.VirtualNode):
super(SuperSequential, self).apply_candidate(abstract_child)
for index in range(len(self)):
for index, module in enumerate(self):
if str(index) in abstract_child:
self.__getitem__(index).apply_candidate(abstract_child[str(index)])
module.apply_candidate(abstract_child[str(index)])
def forward_candidate(self, input):
return self.forward_raw(input)