Skip to content

Commit b7a4b49

Browse files
committed
Add some 384x384 small model weights, 3 variants of mnv4 conv medium on in12k pretrain, and resnetv2-34d on in1k
1 parent facae65 commit b7a4b49

File tree

2 files changed

+21
-0
lines changed

2 files changed

+21
-0
lines changed

timm/models/mobilenetv3.py

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1031,6 +1031,23 @@ def _cfg(url: str = '', **kwargs):
10311031
'mobilenetv4_conv_medium.e500_r224_in1k': _cfg(
10321032
hf_hub_id='timm/',
10331033
crop_pct=0.95, test_input_size=(3, 256, 256), test_crop_pct=1.0, interpolation='bicubic'),
1034+
1035+
'mobilenetv4_conv_medium.e180_r384_in12k': _cfg(
1036+
hf_hub_id='timm/',
1037+
num_classes=11821,
1038+
input_size=(3, 384, 384), pool_size=(12, 12),
1039+
crop_pct=1.0, interpolation='bicubic'),
1040+
'mobilenetv4_conv_medium.e180_ad_r384_in12k': _cfg(
1041+
hf_hub_id='timm/',
1042+
num_classes=11821,
1043+
input_size=(3, 384, 384), pool_size=(12, 12),
1044+
crop_pct=1.0, interpolation='bicubic'),
1045+
'mobilenetv4_conv_medium.e250_r384_in12k': _cfg(
1046+
hf_hub_id='timm/',
1047+
num_classes=11821,
1048+
input_size=(3, 384, 384), pool_size=(12, 12),
1049+
crop_pct=1.0, interpolation='bicubic'),
1050+
10341051
'mobilenetv4_conv_large.e600_r384_in1k': _cfg(
10351052
hf_hub_id='timm/',
10361053
input_size=(3, 384, 384), pool_size=(12, 12),

timm/models/resnetv2.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -710,6 +710,10 @@ def _cfg(url='', **kwargs):
710710
hf_hub_id='timm/',
711711
interpolation='bicubic', crop_pct=0.9, test_input_size=(3, 288, 288), test_crop_pct=1.0,
712712
first_conv='stem.conv1'),
713+
'resnetv2_34d.ra4_e3600_r384_in1k': _cfg(
714+
hf_hub_id='timm/',
715+
interpolation='bicubic', crop_pct=1.0, test_input_size=(3, 448, 448), pool_size=(12, 12),
716+
first_conv='stem.conv1'),
713717
'resnetv2_50.a1h_in1k': _cfg(
714718
hf_hub_id='timm/',
715719
interpolation='bicubic', crop_pct=0.95, test_input_size=(3, 288, 288), test_crop_pct=1.0),

0 commit comments

Comments
 (0)