Trait GradientAggregator 

pub trait GradientAggregator: Send {
    // Required method
    fn aggregate(
        &mut self,
        local_gradients: &[Tensor],
    ) -> Result<Vec<Tensor>, ModelError>;
}

Strategy for combining gradients across distributed workers.

Required Methods

fn aggregate( &mut self, local_gradients: &[Tensor], ) -> Result<Vec<Tensor>, ModelError>

Aggregate local gradients, returning the combined result.

Implementors

impl GradientAggregator for AllReduceAggregator

impl GradientAggregator for LocalAggregator