Struct ViTConfig 

pub struct ViTConfig {
    pub img_size: usize,
    pub patch_size: usize,
    pub in_chans: usize,
    pub embed_dim: usize,
    pub depth: usize,
    pub n_heads: usize,
    pub mlp_ratio: usize,
    pub n_classes: usize,
}

Top-level ViT configuration.

Fields

img_size: usize

Square spatial resolution of the input image (H = W).

patch_size: usize

Patch size (stride = kernel size = patch_size).

in_chans: usize

Number of input channels (e.g. 3 for RGB).

embed_dim: usize

Token embedding dimension.

depth: usize

Number of transformer blocks.

n_heads: usize

Number of attention heads per block.

mlp_ratio: usize

MLP hidden-dim multiplier.

n_classes: usize

Number of output classes for the classification head.

Implementations

impl ViTConfig

pub fn tiny() -> Self

Tiny ViT: suitable for CIFAR-10 style 32×32 RGB images.

img_size=32, patch_size=4, in_chans=3, embed_dim=64, depth=2, n_heads=4, mlp_ratio=4, n_classes=10.

pub fn new( img_size: usize, patch_size: usize, in_chans: usize, embed_dim: usize, depth: usize, n_heads: usize, mlp_ratio: usize, n_classes: usize, ) -> VisionResult<Self>

Validate and construct a ViTConfig.

Errors

• n_classes == 0 → InvalidNumClasses
• patch or image size issues → propagated from PatchEmbedConfig
• embed_dim % n_heads != 0 → HeadDimMismatch

pub fn n_patches(&self) -> usize

Number of non-overlapping patches for this image / patch size.

pub fn seq_len(&self) -> usize

Total sequence length including the CLS token.

Trait Implementations

impl Clone for ViTConfig

fn clone(&self) -> ViTConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for ViTConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl PartialEq for ViTConfig

fn eq(&self, other: &ViTConfig) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl StructuralPartialEq for ViTConfig