Struct ViTEncoderConfig 

pub struct ViTEncoderConfig {
    pub block_cfg: ViTBlockConfig,
    pub depth: usize,
}

Configuration for the ViT encoder stack.

Fields

block_cfg: ViTBlockConfig

Shared configuration for every transformer block.

depth: usize

Number of transformer blocks (encoder depth).

Implementations

impl ViTEncoderConfig

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

Create and validate a ViTEncoderConfig.

Errors

Propagates errors from ViTBlockConfig::new (embed/head validation). Also returns Internal if depth == 0.

Trait Implementations

impl Clone for ViTEncoderConfig

fn clone(&self) -> ViTEncoderConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ViTEncoderConfig

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

Formats the value using the given formatter.

impl PartialEq for ViTEncoderConfig

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