Update NATS-Bench README

This commit is contained in:
D-X-Y
2020-10-29 12:34:04 -07:00
parent 45ce442058
commit e2daffa0ee
2 changed files with 6 additions and 0 deletions

View File

@@ -80,6 +80,10 @@ class NASNetworkDARTS(nn.Module):
for k, op_name in enumerate(self.op_names):
if op_name == 'none': continue
edges.append( (op_name, j, ws[k]) )
# (TODO) xuanyidong:
# Here the selected two edges might come from the same input node.
# And this case could be a problem that two edges will collapse into a single one
# due to our assumption -- at most one edge from an input node during evaluation.
edges = sorted(edges, key=lambda x: -x[-1])
selected_edges = edges[:2]
gene.append( tuple(selected_edges) )