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

updates

Browse Source
This commit is contained in:
D-X-Y
2019-11-11 00:46:02 +11:00
parent fac556c176
commit 7b354d4c74
26 changed files with 1563 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lib/models/__init__.py
View File

@@ -13,7 +13,7 @@ from .cell_searchs import CellStructure, CellArchitectures
# Cell-based NAS Models
def get_cell_based_tiny_net(config):
group_names = ['DARTS-V1', 'DARTS-V2', 'GDAS', 'SETN', 'ENAS']
group_names = ['DARTS-V1', 'DARTS-V2', 'GDAS', 'SETN', 'ENAS', 'RANDOM']
from .cell_searchs import nas_super_nets
if config.name in group_names:
return nas_super_nets[config.name](config.C, config.N, config.max_nodes, config.num_classes, config.space)