Tag along test_vit3 weights

huggingface · Sep 30, 2024 · 545bd40 · 545bd40
1 parent 69b687d
commit 545bd40
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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