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

Wrong hybrid_medium in12k pool sizes

small_things
Ross Wightman 2024-07-25 15:39:21 -07:00
parent 9aa2930760
commit 61df3fde89
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
timm/models/mobilenetv3.py
View File

@ -1020,7 +1020,7 @@ default_cfgs = generate_default_cfgs({
'mobilenetv4_hybrid_medium.e200_r256_in12k_ft_in1k': _cfg(
hf_hub_id='timm/',
input_size=(3, 256, 256), pool_size=(12, 12),
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_r256_in1k': _cfg(
hf_hub_id='timm/',
@ -1036,7 +1036,7 @@ default_cfgs = generate_default_cfgs({
'mobilenetv4_hybrid_medium.e200_r256_in12k': _cfg(
hf_hub_id='timm/',
num_classes=11821,
input_size=(3, 256, 256), pool_size=(12, 12),
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_large.ix_e600_r384_in1k': _cfg(
hf_hub_id='timm/',