Skip to content

Commit 5fe5f9d

Browse files
committed
Add a different mnv4 conv-small weight
1 parent 303f769 commit 5fe5f9d

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

timm/models/mobilenetv3.py

+5
Original file line numberDiff line numberDiff line change
@@ -1024,6 +1024,11 @@ def _cfg(url: str = '', **kwargs):
10241024
'mobilenetv4_conv_small.e1200_r224_in1k': _cfg(
10251025
hf_hub_id='timm/',
10261026
test_input_size=(3, 256, 256), test_crop_pct=0.95, interpolation='bicubic'),
1027+
'mobilenetv4_conv_small.e3600_r256_in1k': _cfg(
1028+
hf_hub_id='timm/',
1029+
mean=IMAGENET_INCEPTION_MEAN, std=IMAGENET_INCEPTION_STD,
1030+
input_size=(3, 256, 256), pool_size=(8, 8), crop_pct=0.95,
1031+
test_input_size=(3, 320, 320), test_crop_pct=1.0, interpolation='bicubic'),
10271032
'mobilenetv4_conv_medium.e500_r256_in1k': _cfg(
10281033
hf_hub_id='timm/',
10291034
input_size=(3, 256, 256), pool_size=(8, 8),

0 commit comments

Comments
 (0)