Struct ModelConfig 

pub struct ModelConfig {
    pub patch_size: usize,
    pub num_queries: usize,
    pub embed_dim: usize,
    pub depth: usize,
    pub num_heads: usize,
    pub mlp_ratio: f64,
    pub num_classes: usize,
    pub drop_path: f64,
    pub norm_eps: f64,
}

Model and runtime configuration for LUNA inference.

ModelConfig mirrors the Python LUNA hyperparameters. Field names match the HuggingFace config.json "model" sub-object.

Fields

patch_size: usize

Patch size in time-samples (default 40).

num_queries: usize

Number of learned cross-attention queries (default 4).

embed_dim: usize

Per-query / per-channel embedding dimension (default 64).

depth: usize

Number of Rotary Transformer encoder blocks (default 8).

num_heads: usize

Number of attention heads per transformer block. Actual head count in the temporal encoder is num_heads * num_queries because the effective dim is embed_dim * num_queries.

mlp_ratio: f64

MLP expansion ratio inside transformer blocks (default 4.0).

num_classes: usize

Number of output classes. 0 = reconstruction (pre-training).

drop_path: f64

Drop-path rate for stochastic depth (default 0.0).

norm_eps: f64

Layer normalisation epsilon (default 1e-5).

Implementations

impl ModelConfig

pub fn hidden_dim(&self) -> usize

Effective hidden dimension after query concatenation: embed_dim * num_queries.

pub fn ffn_hidden_dim(&self) -> usize

FFN hidden dimension inside transformer blocks.

pub fn head_dim(&self) -> usize

Attention head dimension in the temporal encoder.

pub fn total_heads(&self) -> usize

Total number of attention heads in the temporal encoder.

Trait Implementations

impl Clone for ModelConfig

fn clone(&self) -> ModelConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ModelConfig

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

Formats the value using the given formatter.

impl Default for ModelConfig

fn default() -> Self

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for ModelConfig

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.