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

Tag along test_vit3 weights

Browse Source
This commit is contained in:
Ross Wightman 2024-09-30 12:03:32 -07:00
parent 69b687d4cc
commit 545bd4056c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
timm/models/vision_transformer.py
View File

@ -2019,7 +2019,7 @@ default_cfgs = {
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