Struct MitraConfig 

pub struct MitraConfig {
    pub dim: usize,
    pub n_layers: usize,
    pub n_heads: usize,
    pub dim_output: usize,
    pub task: Task,
}

Fields

dim: usize

n_layers: usize

n_heads: usize

dim_output: usize

Max classes the classifier head was trained on (10), or 1 for the regressor.

task: Task

Implementations

impl MitraConfig

pub fn classifier() -> Self

autogluon/mitra-classifier: dim=512, n_layers=12, n_heads=4, dim_output=10.

pub fn regressor() -> Self

autogluon/mitra-regressor: dim=512, n_layers=12, n_heads=4, dim_output=1.

pub fn from_json(v: &Value) -> Result<Self>

Parse the HF config.json shipped alongside the weights: {"dim", "dim_output", "n_layers", "n_heads", "task"}.

Trait Implementations

impl Clone for MitraConfig

fn clone(&self) -> MitraConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for MitraConfig

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

Formats the value using the given formatter.