From aea3b9c8542ec9a19cab544aba9ad458d3cfa8d9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=96=B9=E6=9B=A6?= Date: Thu, 31 Aug 2023 11:41:07 +0800 Subject: [PATCH] Fixing tinyvit input_size issue --- timm/models/tiny_vit.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/timm/models/tiny_vit.py b/timm/models/tiny_vit.py index fc341f9a..c2bc61bb 100644 --- a/timm/models/tiny_vit.py +++ b/timm/models/tiny_vit.py @@ -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) ), })