Skip to content

Commit 61df3fd

Browse files
committed
Wrong hybrid_medium in12k pool sizes
1 parent 9aa2930 commit 61df3fd

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

timm/models/mobilenetv3.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -1020,7 +1020,7 @@ def _cfg(url: str = '', **kwargs):
10201020

10211021
'mobilenetv4_hybrid_medium.e200_r256_in12k_ft_in1k': _cfg(
10221022
hf_hub_id='timm/',
1023-
input_size=(3, 256, 256), pool_size=(12, 12),
1023+
input_size=(3, 256, 256), pool_size=(8, 8),
10241024
crop_pct=0.95, test_input_size=(3, 320, 320), test_crop_pct=1.0, interpolation='bicubic'),
10251025
'mobilenetv4_hybrid_medium.ix_e550_r256_in1k': _cfg(
10261026
hf_hub_id='timm/',
@@ -1036,7 +1036,7 @@ def _cfg(url: str = '', **kwargs):
10361036
'mobilenetv4_hybrid_medium.e200_r256_in12k': _cfg(
10371037
hf_hub_id='timm/',
10381038
num_classes=11821,
1039-
input_size=(3, 256, 256), pool_size=(12, 12),
1039+
input_size=(3, 256, 256), pool_size=(8, 8),
10401040
crop_pct=0.95, test_input_size=(3, 320, 320), test_crop_pct=1.0, interpolation='bicubic'),
10411041
'mobilenetv4_hybrid_large.ix_e600_r384_in1k': _cfg(
10421042
hf_hub_id='timm/',

0 commit comments

Comments
 (0)