FedSGD

Implementation of the [FedSGD17] algorithm.

References

[FedSGD17]

H. Brendan McMahan, Eider Moore, Daniel Ramage, Seth Hampson, and Blaise Aguera y Arcas. Communication-efficient learning of deep networks from decentralized data. In AISTATS (2017). URL: https://arxiv.org/abs/1602.05629

Classes included in fluke.algorithms.fedsgd

FedSGDClient
FedSGD

Classes

class fluke.algorithms.fedsgd.FedSGDClient(index: int, train_set: FastDataLoader, test_set: FastDataLoader, optimizer_cfg: OptimizerConfigurator, loss_fn: Module, local_epochs: int = 3, fine_tuning_epochs: int = 0, **kwargs: dict[str, Any])

Bases: Client

class fluke.algorithms.fedsgd.FedSGD(n_clients: int, data_splitter: DataSplitter, hyper_params: DDict | dict[str, Any], **kwargs: dict[str, Any])

Bases: CentralizedFL