iicd/autodl-projects
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

master -> main

Browse Source
This commit is contained in:
D-X-Y
2021-03-01 21:02:29 +08:00
parent af3ce50a65
commit 493478800c
13 changed files with 32 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lib/nats_bench/api_topology.py
View File

@@ -306,7 +306,7 @@ class NATStopology(NASBenchMetaAPI):
arch_str: the input is a string indicates the architecture topology, such as
|nor_conv_1x1~0|+|none~0|none~1|+|none~0|none~1|skip_connect~2|
search_space: a list of operation string, the default list is the topology search space for NATS-BENCH.
the default value should be be consistent with this line https://github.com/D-X-Y/AutoDL-Projects/blob/master/lib/models/cell_operations.py#L24
the default value should be be consistent with this line https://github.com/D-X-Y/AutoDL-Projects/blob/main/lib/models/cell_operations.py#L24
Returns:
the numpy matrix (2-D np.ndarray) representing the DAG of this architecture topology