Skip to main content

Module gradient_accumulator

Module gradient_accumulator 

Source
Expand description

Gradient accumulator for federated learning rounds.

Accumulates gradients from multiple federated peers across rounds, applying weighted averaging and optional clipping before committing.

Structs§

AccumulatorStats
Running statistics for a GradientAccumulator.
GradientAccumulator
Accumulates gradients from multiple federated peers and produces a weighted-averaged, optionally-clipped aggregate on each commit.
PeerGradient
A gradient vector contributed by a single federated peer.

Enums§

AccumulationError
Errors that can occur during gradient accumulation.
ClipStrategy
Strategy used to clip / clamp gradients before committing.