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