Prototype generic nas model.

lib/models/cell_searchs/__init__.py

@@ -7,6 +7,7 @@ from .search_model_gdas     import TinyNetworkGDAS
 from .search_model_setn     import TinyNetworkSETN
 from .search_model_enas     import TinyNetworkENAS
 from .search_model_random   import TinyNetworkRANDOM
+from .generic_model         import GenericNAS201Model
 from .genotypes             import Structure as CellStructure, architectures as CellArchitectures
 # NASNet-based macro structure
 from .search_model_gdas_nasnet import NASNetworkGDAS
@@ -18,7 +19,8 @@ nas201_super_nets = {'DARTS-V1': TinyNetworkDarts,
                      "GDAS": TinyNetworkGDAS,
                      "SETN": TinyNetworkSETN,
                      "ENAS": TinyNetworkENAS,
-                     "RANDOM": TinyNetworkRANDOM}
+                     "RANDOM": TinyNetworkRANDOM,
+                     "generic": GenericNAS201Model}
 
 nasnet_super_nets = {"GDAS": NASNetworkGDAS,
                      "DARTS": NASNetworkDARTS}