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/nas_201_api/api_201.py
View File

@@ -244,7 +244,7 @@ class NASBench201API(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 search space for NAS-Bench-201
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
:return
the numpy matrix (2-D np.ndarray) representing the DAG of this architecture topology
:usage