Struct ModelStats 

pub struct ModelStats {
    pub total_params: usize,
    pub trainable_params: usize,
    pub num_layers: usize,
    pub d_model: usize,
    pub memory_estimate: usize,
}

Statistics about a transformer model

Fields

total_params: usize

Total number of parameters

trainable_params: usize

Number of trainable parameters

num_layers: usize

Number of layers

d_model: usize

Model dimension

memory_estimate: usize

Memory footprint estimate (bytes)

Implementations

impl ModelStats

pub fn summary(&self) -> String

Format as human-readable string

Trait Implementations

impl Clone for ModelStats

fn clone(&self) -> ModelStats

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ModelStats

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

Formats the value using the given formatter.

impl PartialEq for ModelStats

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