def fbnetc_100(pretrained=False, **kwargs):
""" FBNet-C """
if pretrained:
# pretrained model trained with non-default BN epsilon
kwargs['bn_eps'] = 1e-3
model = _gen_fbnetc(1.0, **kwargs)
if pretrained:
model.load_state_dict(load_state_dict_from_url(model_urls['fbnetc_100']))
return model
def tflite_semnasnet_100(pretrained=False, **kwargs):
""" MNASNet A1, depth multiplier of 1.0. """
# these two args are for compat with tflite pretrained weights
kwargs['folded_bn'] = True
kwargs['padding_same'] = True
model = _gen_mnasnet_a1(1.0, **kwargs)
if pretrained:
model.load_state_dict(load_state_dict_from_url(model_urls['tflite_semnasnet_100']))
return model
def spnasnet_100(pretrained=False, **kwargs):
""" Single-Path NAS Pixel1"""
model = _gen_spnasnet(1.0, **kwargs)
if pretrained:
model.load_state_dict(load_state_dict_from_url(model_urls['spnasnet_100']))
return model
def mnasnet_small(pretrained=False, **kwargs):
""" MNASNet Small, depth multiplier of 1.0. """
model = _gen_mnasnet_small(1.0, **kwargs)
if pretrained:
model.load_state_dict(load_state_dict_from_url(model_urls['mnasnet_small']))
return model
def semnasnet_140(pretrained=False, **kwargs):
""" MNASNet A1 (w/ SE), depth multiplier of 1.4. """
model = _gen_mnasnet_a1(1.4, **kwargs)
if pretrained:
model.load_state_dict(load_state_dict_from_url(model_urls['semnasnet_140']))
return model
def mnasnet_140(pretrained=False, **kwargs):
""" MNASNet B1, depth multiplier of 1.4 """
model = _gen_mnasnet_b1(1.4, **kwargs)
if pretrained:
model.load_state_dict(load_state_dict_from_url(model_urls['mnasnet_140']))
return model
def mnasnet_075(pretrained=False, **kwargs):
""" MNASNet B1, depth multiplier of 0.75. """
model = _gen_mnasnet_b1(0.75, **kwargs)
if pretrained:
model.load_state_dict(load_state_dict_from_url(model_urls['mnasnet_075']))
return model