Skip to content

Commit b751da6

Browse files
committed
Add latest ix (xavier init for mqa) hybrid medium & large weights for MobileNetV4
1 parent d4d4d84 commit b751da6

File tree

1 file changed

+12
-6
lines changed

1 file changed

+12
-6
lines changed

timm/models/mobilenetv3.py

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1018,19 +1018,25 @@ def _cfg(url: str = '', **kwargs):
10181018
input_size=(3, 256, 256), pool_size=(8, 8),
10191019
crop_pct=0.95, test_input_size=(3, 320, 320), test_crop_pct=1.0, interpolation='bicubic'),
10201020

1021-
'mobilenetv4_hybrid_medium.r256': _cfg(
1022-
# hf_hub_id='timm/',
1023-
input_size=(3, 256, 256), pool_size=(8, 8), crop_pct=0.95, interpolation='bicubic'),
1021+
'mobilenetv4_hybrid_medium.ix_e550_r256_in1k': _cfg(
1022+
hf_hub_id='timm/',
1023+
input_size=(3, 256, 256), pool_size=(8, 8),
1024+
crop_pct=0.95, test_input_size=(3, 320, 320), test_crop_pct=1.0, interpolation='bicubic'),
1025+
'mobilenetv4_hybrid_medium.ix_e550_r384_in1k': _cfg(
1026+
hf_hub_id='timm/',
1027+
input_size=(3, 384, 384), pool_size=(12, 12),
1028+
crop_pct=0.95, test_input_size=(3, 448, 448), test_crop_pct=1.0, interpolation='bicubic'),
10241029
'mobilenetv4_hybrid_medium.e500_r224_in1k': _cfg(
10251030
hf_hub_id='timm/',
10261031
crop_pct=0.95, test_input_size=(3, 256, 256), test_crop_pct=1.0, interpolation='bicubic'),
1032+
'mobilenetv4_hybrid_large.ix_e600_r384_in1k': _cfg(
1033+
hf_hub_id='timm/',
1034+
input_size=(3, 384, 384), pool_size=(12, 12),
1035+
crop_pct=0.95, test_input_size=(3, 448, 448), test_crop_pct=1.0, interpolation='bicubic'),
10271036
'mobilenetv4_hybrid_large.e600_r384_in1k': _cfg(
10281037
hf_hub_id='timm/',
10291038
input_size=(3, 384, 384), pool_size=(12, 12),
10301039
crop_pct=0.95, test_input_size=(3, 448, 448), test_crop_pct=1.0, interpolation='bicubic'),
1031-
'mobilenetv4_hybrid_large.r256': _cfg(
1032-
# hf_hub_id='timm/',
1033-
input_size=(3, 256, 256), pool_size=(8, 8), crop_pct=0.95, interpolation='bicubic'),
10341040

10351041
# experimental
10361042
'mobilenetv4_conv_aa_medium.untrained': _cfg(

0 commit comments

Comments
 (0)