Struct StageStats 

pub struct StageStats {
    pub stage_name: String,
    pub tensors_processed: usize,
    pub tensors_skipped: usize,
    pub params_before: usize,
    pub nonzero_params_after: usize,
    pub memory_before_bytes: usize,
    pub memory_after_bytes: usize,
}

Per-stage compression statistics.

Fields

stage_name: String

Name of the stage that produced these stats.

tensors_processed: usize

Number of tensors that were processed by this stage.

tensors_skipped: usize

Number of tensors skipped (e.g. embedding layers with skip_embedding_layers).

params_before: usize

Total number of parameters (elements) entering this stage.

nonzero_params_after: usize

Number of non-zero parameters after this stage.

memory_before_bytes: usize

Total memory (bytes) of all processed tensors before this stage.

memory_after_bytes: usize

Total memory (bytes) of all processed tensors after this stage.

Implementations

impl StageStats

pub fn compression_ratio(&self) -> f32

Ratio of memory_before_bytes / memory_after_bytes. Returns 1.0 if memory_after_bytes is zero to avoid division by zero.

pub fn sparsity(&self) -> f32

Fraction of parameters that are zero after this stage.

Trait Implementations

impl Clone for StageStats

fn clone(&self) -> StageStats

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for StageStats

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

Formats the value using the given formatter.