mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Add some 384x384 small model weights, 3 variants of mnv4 conv medium on in12k pretrain, and resnetv2-34d on in1k

pull/2336/head
Ross Wightman 2024-11-17 11:28:33 -08:00 committed by Ross Wightman
parent facae65947
commit b7a4b49ae6
2 changed files with 21 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
timm/models/mobilenetv3.py
View File

@ -1031,6 +1031,23 @@ default_cfgs = generate_default_cfgs({
'mobilenetv4_conv_medium.e500_r224_in1k': _cfg(
hf_hub_id='timm/',
crop_pct=0.95, test_input_size=(3, 256, 256), test_crop_pct=1.0, interpolation='bicubic'),
'mobilenetv4_conv_medium.e180_r384_in12k': _cfg(
hf_hub_id='timm/',
num_classes=11821,
input_size=(3, 384, 384), pool_size=(12, 12),
crop_pct=1.0, interpolation='bicubic'),
'mobilenetv4_conv_medium.e180_ad_r384_in12k': _cfg(
hf_hub_id='timm/',
num_classes=11821,
input_size=(3, 384, 384), pool_size=(12, 12),
crop_pct=1.0, interpolation='bicubic'),
'mobilenetv4_conv_medium.e250_r384_in12k': _cfg(
hf_hub_id='timm/',
num_classes=11821,
input_size=(3, 384, 384), pool_size=(12, 12),
crop_pct=1.0, interpolation='bicubic'),
'mobilenetv4_conv_large.e600_r384_in1k': _cfg(
hf_hub_id='timm/',
input_size=(3, 384, 384), pool_size=(12, 12),

4
timm/models/resnetv2.py
View File

@ -710,6 +710,10 @@ default_cfgs = generate_default_cfgs({
hf_hub_id='timm/',
interpolation='bicubic', crop_pct=0.9, test_input_size=(3, 288, 288), test_crop_pct=1.0,
first_conv='stem.conv1'),
'resnetv2_34d.ra4_e3600_r384_in1k': _cfg(
hf_hub_id='timm/',
interpolation='bicubic', crop_pct=1.0, test_input_size=(3, 448, 448), pool_size=(12, 12),
first_conv='stem.conv1'),
'resnetv2_50.a1h_in1k': _cfg(
hf_hub_id='timm/',
interpolation='bicubic', crop_pct=0.95, test_input_size=(3, 288, 288), test_crop_pct=1.0),