18 lines
856 B
Python
18 lines
856 B
Python
|
##################################################
|
||
|
|
# Copyright (c) Xuanyi Dong [GitHub D-X-Y], 2019 #
|
||
|
|
##################################################
|
||
|
|
from .search_model_darts_v1 import TinyNetworkDartsV1
|
||
|
|
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,
|
||
|
|
'RANDOM' : TinyNetworkRANDOM}
|