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 latest ix (xavier init for mqa) hybrid medium & large weights for MobileNetV4

pull/2217/head
Ross Wightman 2024-06-24 13:49:55 -07:00
parent d4d4d84fda
commit b751da692d
1 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
timm/models/mobilenetv3.py
View File

@ -1018,19 +1018,25 @@ default_cfgs = generate_default_cfgs({
input_size=(3, 256, 256), pool_size=(8, 8), 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'), crop_pct=0.95, test_input_size=(3, 320, 320), test_crop_pct=1.0, interpolation='bicubic'),
'mobilenetv4_hybrid_medium.r256': _cfg( 'mobilenetv4_hybrid_medium.ix_e550_r256_in1k': _cfg(
# hf_hub_id='timm/', hf_hub_id='timm/',
input_size=(3, 256, 256), pool_size=(8, 8), crop_pct=0.95, interpolation='bicubic'), 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_hybrid_medium.ix_e550_r384_in1k': _cfg(
hf_hub_id='timm/',
input_size=(3, 384, 384), pool_size=(12, 12),
crop_pct=0.95, test_input_size=(3, 448, 448), test_crop_pct=1.0, interpolation='bicubic'),
'mobilenetv4_hybrid_medium.e500_r224_in1k': _cfg( 'mobilenetv4_hybrid_medium.e500_r224_in1k': _cfg(
hf_hub_id='timm/', hf_hub_id='timm/',
crop_pct=0.95, test_input_size=(3, 256, 256), test_crop_pct=1.0, interpolation='bicubic'), crop_pct=0.95, test_input_size=(3, 256, 256), test_crop_pct=1.0, interpolation='bicubic'),
'mobilenetv4_hybrid_large.ix_e600_r384_in1k': _cfg(
hf_hub_id='timm/',
input_size=(3, 384, 384), pool_size=(12, 12),
crop_pct=0.95, test_input_size=(3, 448, 448), test_crop_pct=1.0, interpolation='bicubic'),
'mobilenetv4_hybrid_large.e600_r384_in1k': _cfg( 'mobilenetv4_hybrid_large.e600_r384_in1k': _cfg(
hf_hub_id='timm/', hf_hub_id='timm/',
input_size=(3, 384, 384), pool_size=(12, 12), input_size=(3, 384, 384), pool_size=(12, 12),
crop_pct=0.95, test_input_size=(3, 448, 448), test_crop_pct=1.0, interpolation='bicubic'), crop_pct=0.95, test_input_size=(3, 448, 448), test_crop_pct=1.0, interpolation='bicubic'),
'mobilenetv4_hybrid_large.r256': _cfg(
# hf_hub_id='timm/',
input_size=(3, 256, 256), pool_size=(8, 8), crop_pct=0.95, interpolation='bicubic'),
# experimental # experimental
'mobilenetv4_conv_aa_medium.untrained': _cfg( 'mobilenetv4_conv_aa_medium.untrained': _cfg(