Struct ModelConfig 

pub struct ModelConfig {
    pub model_size: ModelSize,
    pub feature_size: usize,
    pub num_heads: usize,
    pub num_layers: usize,
    pub dim_feedforward: usize,
    pub num_global_tokens: usize,
    pub global_token_layer: usize,
    pub num_channels: usize,
    pub patch_size: usize,
    pub conv_channels: [usize; 3],
    pub norm_groups: [usize; 3],
    pub layer_norm_eps: f64,
}

Full model configuration derived from model size.

All values match the Python EEG-DINO defaults exactly.

Fields

model_size: ModelSize

feature_size: usize

Embedding dimension (d_model): 200 / 512 / 1024

num_heads: usize

Number of attention heads: 8 / 16 / 24

num_layers: usize

Number of transformer encoder layers: 12 / 12 / 24

dim_feedforward: usize

Feed-forward hidden dimension: 512 / 1024 / 2048

num_global_tokens: usize

Number of learnable global tokens (default: 1)

global_token_layer: usize

Layer index (1-based) at which global tokens are injected (default: 1)

num_channels: usize

Number of EEG channels (default: 19)

patch_size: usize

Samples per patch (default: 200)

conv_channels: [usize; 3]

Conv channel widths for the 3 conv layers in proj_in: [c1, c2, c3]

norm_groups: [usize; 3]

GroupNorm group counts for the 3 norm layers in proj_in

layer_norm_eps: f64

LayerNorm epsilon

Implementations

impl ModelConfig

pub fn from_size(size: ModelSize) -> Self

pub fn from_file(path: &Path) -> Result<Self>

Load config from a JSON file.

pub fn head_dim(&self) -> usize

Head dimension (feature_size / num_heads).

pub fn spectral_bins(&self) -> usize

Number of spectral bins from rfft of patch_size samples: patch_size/2 + 1.

pub fn temporal_conv_out(&self) -> usize

Temporal output dimension of the conv stack: floor((patch_size - 49 + 2*24) / 25) + 1.

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<'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.

impl Serialize for ModelConfig

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.