mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Add a different mnv4 conv-small weight

Browse Source
This commit is contained in:
Ross Wightman 2024-12-02 16:14:37 -08:00
parent 303f7691a1
commit 5fe5f9d488
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
timm/models/mobilenetv3.py
View File

@ -1024,6 +1024,11 @@ default_cfgs = generate_default_cfgs({
'mobilenetv4_conv_small.e1200_r224_in1k': _cfg(
hf_hub_id='timm/',
test_input_size=(3, 256, 256), test_crop_pct=0.95, interpolation='bicubic'),
'mobilenetv4_conv_small.e3600_r256_in1k': _cfg(
hf_hub_id='timm/',
mean=IMAGENET_INCEPTION_MEAN, std=IMAGENET_INCEPTION_STD,
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_conv_medium.e500_r256_in1k': _cfg(
hf_hub_id='timm/',
input_size=(3, 256, 256), pool_size=(8, 8),