Trait GroupedOptimizer 

pub trait GroupedOptimizer<A: Float + ScalarOperand + Debug, D: Dimension>: Optimizer<A, D> {
    // Required methods
    fn add_group(
        &mut self,
        params: Vec<Array<A, D>>,
        config: ParameterGroupConfig<A>,
    ) -> Result<usize>;
    fn get_group(&self, groupid: usize) -> Result<&ParameterGroup<A, D>>;
    fn get_group_mut(
        &mut self,
        groupid: usize,
    ) -> Result<&mut ParameterGroup<A, D>>;
    fn groups(&self) -> &[ParameterGroup<A, D>];
    fn groups_mut(&mut self) -> &mut [ParameterGroup<A, D>];
    fn step_group(
        &mut self,
        group_id: usize,
        gradients: &[Array<A, D>],
    ) -> Result<Vec<Array<A, D>>>;
    fn set_group_learning_rate(&mut self, groupid: usize, lr: A) -> Result<()>;
    fn set_group_weight_decay(&mut self, groupid: usize, wd: A) -> Result<()>;
}

Optimizer with parameter group support

Required Methods

fn add_group( &mut self, params: Vec<Array<A, D>>, config: ParameterGroupConfig<A>, ) -> Result<usize>

Add a parameter group

fn get_group(&self, groupid: usize) -> Result<&ParameterGroup<A, D>>

Get parameter group by ID

fn get_group_mut(&mut self, groupid: usize) -> Result<&mut ParameterGroup<A, D>>

Get mutable parameter group by ID

fn groups(&self) -> &[ParameterGroup<A, D>]

Get all parameter groups

fn groups_mut(&mut self) -> &mut [ParameterGroup<A, D>]

Get all parameter groups mutably

fn step_group( &mut self, group_id: usize, gradients: &[Array<A, D>], ) -> Result<Vec<Array<A, D>>>

Step for a specific group

fn set_group_learning_rate(&mut self, groupid: usize, lr: A) -> Result<()>

Set learning rate for a specific group

fn set_group_weight_decay(&mut self, groupid: usize, wd: A) -> Result<()>

Set weight decay for a specific group

Implementors

impl<A: Float + ScalarOperand + Debug + Send + Sync, D: Dimension + Send + Sync> GroupedOptimizer<A, D> for GroupedAdam<A, D>