p2pfl.learning.aggregators.fedavg module

Federated Averaging (FedAvg) Aggregator.

class p2pfl.learning.aggregators.fedavg.FedAvg(disable_partial_aggregation=False)

Bases: WeightAggregator

Federated Averaging (FedAvg) [McMahan et al., 2016].

Inherits from WeightAggregator as FedAvg works with neural network weight tensors that can be averaged.

Paper: https://arxiv.org/abs/1602.05629.

Parameters:

disable_partial_aggregation (bool)

SUPPORTS_PARTIAL_AGGREGATION: bool = True

addr: str

partial_aggregation: bool