Update NATS-Bench README
This commit is contained in:
		| @@ -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) ) | ||||
|   | ||||
		Reference in New Issue
	
	Block a user