update TF

This commit is contained in:
D-X-Y
2020-01-20 15:24:12 +11:00
parent 1ce3249a5a
commit 533a508444
5 changed files with 339 additions and 2 deletions

View File

@@ -9,7 +9,7 @@ __all__ = ['get_cell_based_tiny_net', 'get_search_spaces']
# the cell-based NAS models
def get_cell_based_tiny_net(config):
group_names = ['GDAS']
group_names = ['GDAS', 'DARTS']
if config.name in group_names:
from .cell_searchs import nas_super_nets
from .cell_operations import SearchSpaceNames