Struct ParameterGroupConfig 

pub struct ParameterGroupConfig<A: Float> {
    pub learning_rate: Option<A>,
    pub weight_decay: Option<A>,
    pub momentum: Option<A>,
    pub constraints: Vec<ParameterConstraint<A>>,
    pub custom_params: HashMap<String, A>,
}

Configuration for a parameter group

Fields

learning_rate: Option<A>

Learning rate for this group

weight_decay: Option<A>

Weight decay for this group

momentum: Option<A>

Momentum for this group (if applicable)

constraints: Vec<ParameterConstraint<A>>

Parameter constraints for this group

custom_params: HashMap<String, A>

Custom parameters as key-value pairs

Implementations

impl<A: Float + Send + Sync> ParameterGroupConfig<A>

pub fn new() -> Self

Create a new parameter group configuration

pub fn with_learning_rate(self, lr: A) -> Self

Set learning rate

pub fn with_weight_decay(self, wd: A) -> Self

Set weight decay

pub fn with_momentum(self, momentum: A) -> Self

Set momentum

pub fn with_custom_param(self, key: String, value: A) -> Self

Add custom parameter

pub fn with_constraint(self, constraint: ParameterConstraint<A>) -> Self

Add a parameter constraint

pub fn with_value_clip(self, min: A, max: A) -> Self

Add value clipping constraint

pub fn with_l2_norm_constraint(self, maxnorm: A) -> Self

Add L2 norm constraint

pub fn with_l1_norm_constraint(self, maxnorm: A) -> Self

Add L1 norm constraint

pub fn with_non_negative(self) -> Self

Add non-negativity constraint

pub fn with_unit_sphere(self) -> Self

Add unit sphere constraint

pub fn with_simplex(self) -> Self

Add simplex constraint (sum to 1, all non-negative)

pub fn with_orthogonal(self, tolerance: A) -> Self

Add orthogonal constraint for matrices

pub fn with_positive_definite(self, mineigenvalue: A) -> Self

Add positive definite constraint for symmetric matrices

pub fn with_spectral_norm(self, maxnorm: A) -> Self

Add spectral norm constraint

pub fn with_nuclear_norm(self, maxnorm: A) -> Self

Add nuclear norm constraint

pub fn with_custom_constraint(self, name: String) -> Self

Add custom constraint

Trait Implementations

impl<A: Clone + Float> Clone for ParameterGroupConfig<A>

fn clone(&self) -> ParameterGroupConfig<A>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<A: Debug + Float> Debug for ParameterGroupConfig<A>

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

Formats the value using the given formatter.

impl<A: Float + Send + Sync> Default for ParameterGroupConfig<A>

fn default() -> Self

Returns the “default value” for a type.