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

Fixing tinyvit input_size issue

Browse Source
This commit is contained in:
方曦 2023-08-31 11:41:07 +08:00 committed by Ross Wightman
parent fabc4e5bcd
commit aea3b9c854
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
timm/models/tiny_vit.py
View File

@ -624,6 +624,7 @@ def _cfg(url='', **kwargs):
'classifier': 'head.fc',
'fixed_input_size': True,
'pool_size': None,
'input_size': (3, 224, 224),
**kwargs,
}
@ -660,10 +661,12 @@ default_cfgs = generate_default_cfgs({
url='https://github.com/wkcn/TinyViT-model-zoo/releases/download/checkpoints/tiny_vit_21m_1k.pth'
),
'tiny_vit_21m_384.dist_in22k_ft_in1k': _cfg(
url='https://github.com/wkcn/TinyViT-model-zoo/releases/download/checkpoints/tiny_vit_21m_22kto1k_384_distill.pth'
url='https://github.com/wkcn/TinyViT-model-zoo/releases/download/checkpoints/tiny_vit_21m_22kto1k_384_distill.pth',
input_size=(3, 384, 384)
),
'tiny_vit_21m_512.dist_in22k_ft_in1k': _cfg(
url='https://github.com/wkcn/TinyViT-model-zoo/releases/download/checkpoints/tiny_vit_21m_22kto1k_512_distill.pth'
url='https://github.com/wkcn/TinyViT-model-zoo/releases/download/checkpoints/tiny_vit_21m_22kto1k_512_distill.pth',
input_size=(3, 512, 512)
),
})