From 5fe5f9d48880fa1ec4bd28e1dade332b6fba0988 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Mon, 2 Dec 2024 16:14:37 -0800 Subject: [PATCH] Add a different mnv4 conv-small weight --- timm/models/mobilenetv3.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/timm/models/mobilenetv3.py b/timm/models/mobilenetv3.py index c5e81391f..79a51f772 100644 --- a/timm/models/mobilenetv3.py +++ b/timm/models/mobilenetv3.py @@ -1024,6 +1024,11 @@ def _cfg(url: str = '', **kwargs): 'mobilenetv4_conv_small.e1200_r224_in1k': _cfg( hf_hub_id='timm/', test_input_size=(3, 256, 256), test_crop_pct=0.95, interpolation='bicubic'), + 'mobilenetv4_conv_small.e3600_r256_in1k': _cfg( + hf_hub_id='timm/', + mean=IMAGENET_INCEPTION_MEAN, std=IMAGENET_INCEPTION_STD, + input_size=(3, 256, 256), pool_size=(8, 8), crop_pct=0.95, + test_input_size=(3, 320, 320), test_crop_pct=1.0, interpolation='bicubic'), 'mobilenetv4_conv_medium.e500_r256_in1k': _cfg( hf_hub_id='timm/', input_size=(3, 256, 256), pool_size=(8, 8),