Struct ModelConfig 

pub struct ModelConfig {

    pub encoder_name: String,
    pub num_leads: usize,
    pub patch_size_time: usize,
    pub patch_size_ch: usize,
    pub lead_wise: usize,
    pub sample_rate: usize,
    pub window_size_sec: usize,
    pub seq_len: usize,
    pub width: usize,
    pub depth: usize,
    pub heads: usize,
    pub mlp_dim: usize,
    pub dim_head: usize,
}

Model and data configuration for OSF inference.

ModelConfig mirrors the Python OSF ViT hyperparameters stored in the metadata dict of osf_backbone.pth.

Fields

encoder_name: String

Model variant name (e.g. “vit_base”).

num_leads: usize

Number of input channels (default 12).

patch_size_time: usize

Temporal patch size in samples (default 64).

patch_size_ch: usize

Channel patch size (default 4).

lead_wise: usize

0 = 1D patchify (all channels), 1 = 2D patchify (channel groups).

sample_rate: usize

Sampling rate (Hz, default 64).

window_size_sec: usize

Window size in seconds (default 30).

seq_len: usize

Sequence length = sample_rate * window_size_sec (default 1920).

width: usize

Transformer hidden dimension (default 768).

depth: usize

Number of transformer blocks (default 12).

heads: usize

Number of attention heads (default 12).

mlp_dim: usize

MLP hidden dimension (default 3072 = 4 * width).

dim_head: usize

Attention head dimension (default 64 = width / heads).

Implementations

impl ModelConfig

pub fn num_patches_time(&self) -> usize

Number of time patches per channel row.

pub fn num_lead_rows(&self) -> usize

Number of channel rows (lead groups).

pub fn num_patches(&self) -> usize

Total number of patches (excluding CLS token).

pub fn for_variant(name: &str) -> Self

Build config for a specific variant.

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.

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.