maxvit_large¶

lucid.models.maxvit_large(in_channels: int = 3, num_classes: int = 1000, **kwargs) → MaxViT¶

The maxvit_large function constructs the large MaxViT preset.

Function Signature¶

@register_model
def maxvit_large(in_channels: int = 3, num_classes: int = 1000, **kwargs) -> MaxViT

Parameters¶

in_channels (int, optional): Number of input channels. Default is 3.
num_classes (int, optional): Number of output classes for classification. Default is 1000.
kwargs (dict, optional): Additional keyword arguments forwarded to MaxViTConfig, excluding the preset depths, channels, and embed_dim fields.

Returns¶

MaxViT: A MaxViT instance constructed from the large preset config.

Preset Config¶

depths: (2, 6, 14, 2)
channels: (128, 256, 512, 1024)
embed_dim: 128