From 545bd4056cf49a1dfbf77e52529fac30c134e916 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Mon, 30 Sep 2024 12:03:32 -0700 Subject: [PATCH] Tag along test_vit3 weights --- timm/models/vision_transformer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/timm/models/vision_transformer.py b/timm/models/vision_transformer.py index 2c9c7f4a5..c0a28085d 100644 --- a/timm/models/vision_transformer.py +++ b/timm/models/vision_transformer.py @@ -2019,7 +2019,7 @@ def _cfg(url: str = '', **kwargs) -> Dict[str, Any]: hf_hub_id='timm/', input_size=(3, 160, 160), crop_pct=0.95), 'test_vit3.r160_in1k': _cfg( - #hf_hub_id='timm/', + hf_hub_id='timm/', input_size=(3, 160, 160), crop_pct=0.95), } @@ -3238,7 +3238,7 @@ def test_vit3(pretrained: bool = False, **kwargs) -> VisionTransformer: """ ViT Test """ model_args = dict( - patch_size=16, embed_dim=96, depth=10, num_heads=3, mlp_ratio=2, + patch_size=16, embed_dim=96, depth=9, num_heads=3, mlp_ratio=2, class_token=False, reg_tokens=1, global_pool='map', init_values=1e-5) model = _create_vision_transformer('test_vit3', pretrained=pretrained, **dict(model_args, **kwargs)) return model