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

View File

@@ -3,10 +3,12 @@ from .search_model_darts_v2 import TinyNetworkDartsV2
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 .genotypes import Structure as CellStructure, architectures as CellArchitectures
nas_super_nets = {'DARTS-V1': TinyNetworkDartsV1,
'DARTS-V2': TinyNetworkDartsV2,
'GDAS' : TinyNetworkGDAS,
'SETN' : TinyNetworkSETN,
'ENAS' : TinyNetworkENAS}
'ENAS' : TinyNetworkENAS,
'RANDOM' : TinyNetworkRANDOM}