iicd/xautodl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update TF

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

4
lib/tf_models/cell_searchs/__init__.py
View File

@@ -2,5 +2,7 @@
# Copyright (c) Xuanyi Dong [GitHub D-X-Y], 2019 #
##################################################
from .search_model_gdas import TinyNetworkGDAS
from .search_model_darts import TinyNetworkDARTS
nas_super_nets = {'GDAS': TinyNetworkGDAS}
nas_super_nets = {'GDAS' : TinyNetworkGDAS,
'DARTS': TinyNetworkDARTS}