Struct ParamGroup 

pub struct ParamGroup {
    pub params: Vec<Arc<RwLock<Tensor>>>,
    pub lr: f32,
    pub options: HashMap<String, f32>,
}

Parameter group

Fields

params: Vec<Arc<RwLock<Tensor>>>

lr: f32

options: HashMap<String, f32>

Implementations

impl ParamGroup

pub fn new(params: Vec<Arc<RwLock<Tensor>>>, lr: f32) -> Self

pub fn with_options(self, options: HashMap<String, f32>) -> Self

pub fn add_param(&mut self, param: Arc<RwLock<Tensor>>)

Add a single parameter to the group

pub fn get_option(&self, key: &str, default: f32) -> f32

Get a specific option value, falling back to a default

pub fn set_option(&mut self, key: String, value: f32)

Set a specific option value

pub fn param_count(&self) -> usize

Get the number of parameters in this group

pub fn is_empty(&self) -> bool

Check if this group has any parameters

pub fn params_with_grads(&self) -> Vec<&Arc<RwLock<Tensor>>>

Get all parameters that have gradients

pub fn validate(&self) -> bool

Validate that all parameters in the group are valid

pub fn get_shape_counts(&self) -> HashMap<Vec<usize>, usize>

Get parameter count for each unique shape in the group

pub fn total_param_count(&self) -> usize

Get total number of parameters (not tensors, but individual parameters)

pub fn zero_grad(&self)

Clear gradients for all parameters in this group

pub fn has_any_grads(&self) -> bool

Check if any parameter in the group has gradients

pub fn grad_norm(&self) -> Result<f32>

Get gradient norm for all parameters in the group

pub fn clip_grads(&self, max_norm: f32) -> Result<f32>

Apply gradient clipping to all parameters in the group

Trait Implementations

impl Clone for ParamGroup

fn clone(&self) -> ParamGroup

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ParamGroup

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

Formats the value using the given formatter.