Skip to content
Snippets Groups Projects
Unverified Commit be7e5725 authored by Patrick Labatut's avatar Patrick Labatut Committed by GitHub
Browse files

Add missing hubconf arg (#178)

Add missing explicit layers argument
parent 10d42014
Branches
No related tags found
No related merge requests found
...@@ -159,24 +159,24 @@ def dinov2_vits14_lc(*, layers: int = 4, pretrained: bool = True, **kwargs): ...@@ -159,24 +159,24 @@ def dinov2_vits14_lc(*, layers: int = 4, pretrained: bool = True, **kwargs):
return _make_dinov2_linear_classifier(arch_name="vit_small", layers=layers, pretrained=pretrained, **kwargs) return _make_dinov2_linear_classifier(arch_name="vit_small", layers=layers, pretrained=pretrained, **kwargs)
def dinov2_vitb14_lc(*, pretrained: bool = True, **kwargs): def dinov2_vitb14_lc(*, layers: int = 4, pretrained: bool = True, **kwargs):
""" """
Linear classifier (1 or 4 layers) on top of a DINOv2 ViT-B/14 backbone (optionally) pretrained on the LVD-142M dataset and trained on ImageNet-1k. Linear classifier (1 or 4 layers) on top of a DINOv2 ViT-B/14 backbone (optionally) pretrained on the LVD-142M dataset and trained on ImageNet-1k.
""" """
return _make_dinov2_linear_classifier(arch_name="vit_base", pretrained=pretrained, **kwargs) return _make_dinov2_linear_classifier(arch_name="vit_base", layers=layers, pretrained=pretrained, **kwargs)
def dinov2_vitl14_lc(*, pretrained: bool = True, **kwargs): def dinov2_vitl14_lc(*, layers: int = 4, pretrained: bool = True, **kwargs):
""" """
Linear classifier (1 or 4 layers) on top of a DINOv2 ViT-L/14 backbone (optionally) pretrained on the LVD-142M dataset and trained on ImageNet-1k. Linear classifier (1 or 4 layers) on top of a DINOv2 ViT-L/14 backbone (optionally) pretrained on the LVD-142M dataset and trained on ImageNet-1k.
""" """
return _make_dinov2_linear_classifier(arch_name="vit_large", pretrained=pretrained, **kwargs) return _make_dinov2_linear_classifier(arch_name="vit_large", layers=layers, pretrained=pretrained, **kwargs)
def dinov2_vitg14_lc(*, pretrained: bool = True, **kwargs): def dinov2_vitg14_lc(*, layers: int = 4, pretrained: bool = True, **kwargs):
""" """
Linear classifier (1 or 4 layers) on top of a DINOv2 ViT-g/14 backbone (optionally) pretrained on the LVD-142M dataset and trained on ImageNet-1k. Linear classifier (1 or 4 layers) on top of a DINOv2 ViT-g/14 backbone (optionally) pretrained on the LVD-142M dataset and trained on ImageNet-1k.
""" """
return _make_dinov2_linear_classifier( return _make_dinov2_linear_classifier(
arch_name="vit_giant2", ffn_layer="swiglufused", pretrained=pretrained, **kwargs arch_name="vit_giant2", layers=layers, ffn_layer="swiglufused", pretrained=pretrained, **kwargs
) )
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment