File size: 885 Bytes
188f311 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
# Once for All: Train One Network and Specialize it for Efficient Deployment
# Han Cai, Chuang Gan, Tianzhe Wang, Zhekai Zhang, Song Han
# International Conference on Learning Representations (ICLR), 2020.
from .proxyless_nets import *
from .mobilenet_v3 import *
from .resnets import *
from .wide_resnet import WideResNet
from .resnet_trades import *
def get_net_by_name(name):
if name == ProxylessNASNets.__name__:
return ProxylessNASNets
elif name == MobileNetV3.__name__:
return MobileNetV3
elif name == ResNets.__name__:
return ResNets
if name == ProxylessNASNets_Cifar.__name__:
return ProxylessNASNets_Cifar
elif name == MobileNetV3_Cifar.__name__:
return MobileNetV3
elif name == ResNets_Cifar.__name__:
return ResNets_Cifar
else:
raise ValueError("unrecognized type of network: %s" % name)
|