Struct ModelParameters 

pub struct ModelParameters {
    pub parameter_count: usize,
    pub parameter_names: Vec<String>,
    pub parameter_shapes: HashMap<String, Vec<usize>>,
    pub total_memory_bytes: usize,
}

Model parameters for ZeRO-3 initialization

Fields

parameter_count: usize

parameter_names: Vec<String>

parameter_shapes: HashMap<String, Vec<usize>>

total_memory_bytes: usize

Implementations

impl ModelParameters

pub fn new() -> Self

Create a new empty model parameters collection

pub fn add_parameter(&mut self, name: String, shape: Vec<usize>)

Add a parameter to the collection

pub fn has_parameter(&self, name: &str) -> bool

Check if a parameter exists

pub fn add_parameter_with_size( &mut self, name: String, shape: Vec<usize>, element_size: usize, )

Add a parameter with custom element size

pub fn get_parameter_shape(&self, name: &str) -> Option<&Vec<usize>>

Get the shape of a parameter by name

pub fn get_parameter_size(&self, name: &str) -> Option<usize>

Get the number of elements in a parameter

pub fn total_parameters(&self) -> usize

Get total number of parameters

pub fn memory_usage_mb(&self) -> f64

Get memory usage in MB

pub fn get_statistics(&self) -> ModelParameterStats

Get parameter statistics

Trait Implementations

impl Debug for ModelParameters

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

Formats the value using the given formatter.

impl Default for ModelParameters

fn default() -> Self

Returns the “default value” for a type.