Trait GradientAggregator 

pub trait GradientAggregator {
    // Required method
    fn aggregate(
        &self,
        gradients: &[HashMap<String, Value>],
    ) -> Result<HashMap<String, Value>>;
}

Contract for gradient aggregation across workers.

Required Methods

fn aggregate( &self, gradients: &[HashMap<String, Value>], ) -> Result<HashMap<String, Value>>

Combine per-worker gradients (keyed by node id) into the one set every worker then applies.

Implementations on Foreign Types

impl GradientAggregator for GradientAggregation

fn aggregate( &self, gradients: &[HashMap<String, Value>], ) -> Result<HashMap<String, Value>>

Implementors