vit_small¶

lucid.models.vit_small(image_size: int = 224, patch_size: int = 16, num_classes: int = 1000, **kwargs) → ViT¶

The vit_small function instantiates a small Vision Transformer (ViT-S) model with a predefined architecture. This model is a balance between efficiency and performance, making it suitable for mid-scale vision tasks.

Total Parameters: 22,050,664

Function Signature¶

@register_model
def vit_small(
    image_size: int = 224,
    patch_size: int = 16,
    num_classes: int = 1000,
    **kwargs
) -> ViT

Parameters¶

image_size (int, optional): The size of the input image (assumes square images). Default is 224.
patch_size (int, optional): The size of the patches the image is divided into. Default is 16.
num_classes (int, optional): The number of output classes for classification. Default is 1000.
kwargs (dict, optional): Additional parameters for customization.

Returns¶

ViT: An instance of the ViT class configured as a small vision transformer.

Examples¶

>>> import lucid.models as models
>>> model = models.vit_small()
>>> print(model)
ViT(img_size=224, patch_size=16, num_classes=1000, ...)