convnext_v2_atto

lucid.models.convnext_v2_atto(num_classes: int = 1000, **kwargs) → ConvNeXt_V2

The convnext_v2_atto creates a ConvNeXt_V2 variant with an ultra-compact configuration, designed for highly resource-constrained environments. This model variant optimizes computational efficiency while maintaining competitive performance for small-scale image classification tasks.

Total Parameters: 3,708,400

Function Signature

@register_model
def convnext_v2_atto(num_classes: int = 1000, **kwargs) -> ConvNeXt_V2

Parameters

  • num_classes (int, optional): The number of output classes for classification. Default is 1000.

  • kwargs: Additional keyword arguments for customizing the ConvNeXt-v2 architecture.

Returns

  • model (ConvNeXt_V2): An instance of the ConvNeXt-v2 class configured as the atto variant.

Examples

Basic Example

import lucid.models as models

# Create convnext_v2_atto with default 1000 classes
model = models.convnext_v2_atto(num_classes=1000)

# Input tensor with shape (1, 3, 224, 224)
input_ = lucid.random.randn(1, 3, 224, 224)

# Perform forward pass
output = model(input_)

print(output.shape)  # Shape: (1, 1000)