Skip to content

Commit 7893312

Browse files
committed
Fix silly typo
1 parent 708d87a commit 7893312

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

timm/models/vision_transformer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -683,7 +683,7 @@ def vit_large_patch16_384(pretrained=False, **kwargs):
683683
def vit_base_patch16_sam_224(pretrained=False, **kwargs):
684684
""" ViT-Base (ViT-B/16) w/ SAM pretrained weights. Paper: https://arxiv.org/abs/2106.01548
685685
"""
686-
# NOTE original SAM weights releaes worked with representation_size=768
686+
# NOTE original SAM weights release worked with representation_size=768
687687
model_kwargs = dict(patch_size=16, embed_dim=768, depth=12, num_heads=12, representation_size=0, **kwargs)
688688
model = _create_vision_transformer('vit_base_patch16_sam_224', pretrained=pretrained, **model_kwargs)
689689
return model
@@ -693,7 +693,7 @@ def vit_base_patch16_sam_224(pretrained=False, **kwargs):
693693
def vit_base_patch32_sam_224(pretrained=False, **kwargs):
694694
""" ViT-Base (ViT-B/32) w/ SAM pretrained weights. Paper: https://arxiv.org/abs/2106.01548
695695
"""
696-
# NOTE original SAM weights releaes worked with representation_size=768
696+
# NOTE original SAM weights release worked with representation_size=768
697697
model_kwargs = dict(patch_size=32, embed_dim=768, depth=12, num_heads=12, representation_size=0, **kwargs)
698698
model = _create_vision_transformer('vit_base_patch32_sam_224', pretrained=pretrained, **model_kwargs)
699699
return model

0 commit comments

Comments
 (0)