Skip to content

Commit 76b0e99

Browse files
committed
Placeholder for new mnv3 model
1 parent ed7aaf8 commit 76b0e99

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

timm/models/mobilenetv3.py

+11-2
Original file line numberDiff line numberDiff line change
@@ -453,7 +453,8 @@ def _gen_mobilenet_v3_rw(
453453

454454

455455
def _gen_mobilenet_v3(
456-
variant: str, channel_multiplier: float = 1.0, group_size=None, pretrained: bool = False, **kwargs
456+
variant: str, channel_multiplier: float = 1.0, depth_multiplier: float = 1.0,
457+
group_size=None, pretrained: bool = False, **kwargs
457458
) -> MobileNetV3:
458459
"""Creates a MobileNet-V3 model.
459460
@@ -537,7 +538,7 @@ def _gen_mobilenet_v3(
537538
]
538539
se_layer = partial(SqueezeExcite, gate_layer='hard_sigmoid', force_act_layer=nn.ReLU, rd_round_fn=round_channels)
539540
model_kwargs = dict(
540-
block_args=decode_arch_def(arch_def, group_size=group_size),
541+
block_args=decode_arch_def(arch_def, depth_multiplier=depth_multiplier, group_size=group_size),
541542
num_features=num_features,
542543
stem_size=16,
543544
fix_stem=channel_multiplier < 0.75,
@@ -927,6 +928,9 @@ def _cfg(url: str = '', **kwargs):
927928
origin_url='https://github.com/Alibaba-MIIL/ImageNet21K',
928929
paper_ids='arXiv:2104.10972v4',
929930
interpolation='bilinear', mean=(0., 0., 0.), std=(1., 1., 1.), num_classes=11221),
931+
'mobilenetv3_large_150d.untrained': _cfg(
932+
#hf_hub_id='timm/',
933+
),
930934

931935
'mobilenetv3_small_050.lamb_in1k': _cfg(
932936
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/mobilenetv3_small_050_lambc-4b7bbe87.pth',
@@ -1099,6 +1103,11 @@ def mobilenetv3_large_100(pretrained: bool = False, **kwargs) -> MobileNetV3:
10991103
model = _gen_mobilenet_v3('mobilenetv3_large_100', 1.0, pretrained=pretrained, **kwargs)
11001104
return model
11011105

1106+
@register_model
1107+
def mobilenetv3_large_150d(pretrained: bool = False, **kwargs) -> MobileNetV3:
1108+
""" MobileNet V3 """
1109+
model = _gen_mobilenet_v3('mobilenetv3_large_150d', 1.5, depth_multiplier=1.2, pretrained=pretrained, **kwargs)
1110+
return model
11021111

11031112
@register_model
11041113
def mobilenetv3_small_050(pretrained: bool = False, **kwargs) -> MobileNetV3:

0 commit comments

Comments
 (0)