Struct MaeConfig 

pub struct MaeConfig {
    pub img_size: usize,
    pub patch_size: usize,
    pub in_channels: usize,
    pub encoder_dim: usize,
    pub encoder_depth: usize,
    pub encoder_heads: usize,
    pub decoder_dim: usize,
    pub decoder_depth: usize,
    pub decoder_heads: usize,
    pub mlp_ratio: usize,
    pub mask_ratio: f32,
}

Configuration for an MAE (Masked Autoencoder) model.

Fields

img_size: usize

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

patch_size: usize

Patch size (must divide img_size).

in_channels: usize

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

encoder_dim: usize

Encoder embedding dimension.

encoder_depth: usize

Number of encoder transformer blocks.

encoder_heads: usize

Number of encoder attention heads.

decoder_dim: usize

Decoder embedding dimension (usually smaller than encoder_dim).

decoder_depth: usize

Number of decoder transformer blocks.

decoder_heads: usize

Number of decoder attention heads.

mlp_ratio: usize

MLP hidden-dim multiplier (shared by encoder + decoder blocks).

mask_ratio: f32

Fraction of patches to mask, in [0, 1] (e.g. 0.75).

Implementations

impl MaeConfig

pub fn new( img_size: usize, patch_size: usize, in_channels: usize, encoder_dim: usize, encoder_depth: usize, encoder_heads: usize, decoder_dim: usize, decoder_depth: usize, decoder_heads: usize, mlp_ratio: usize, mask_ratio: f32, ) -> VisionResult<Self>

Build and validate a new MaeConfig.

Errors

• img_size % patch_size != 0 → InvalidPatchSize
• any zero-sized dimension → InvalidEmbedDim / EmptyInput
• mask_ratio outside [0, 1] → Internal
• encoder_dim % encoder_heads != 0 or decoder_dim % decoder_heads != 0 → HeadDimMismatch

pub fn n_patches(&self) -> usize

Number of non-overlapping image patches.

pub fn patch_pixels(&self) -> usize

Per-patch flat pixel count (patch_size² · in_channels).

Trait Implementations

impl Clone for MaeConfig

fn clone(&self) -> MaeConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for MaeConfig

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

Formats the value using the given formatter.

impl PartialEq for MaeConfig

fn eq(&self, other: &MaeConfig) -> 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 MaeConfig