update TF

2020-01-20 15:24:12 +11:00
parent 1ce3249a5a
commit 533a508444
5 changed files with 339 additions and 2 deletions
--- a/lib/tf_models/cell_searchs/init.py
+++ b/lib/tf_models/cell_searchs/init.py
@@ -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}