/federated_learning_pytorch

Simplified implementation of federated learning in PyTorch

Primary LanguageJupyter NotebookMIT LicenseMIT

Federated Learning PyTorch

Simplified implementation of Federated Learning in PyTorch

Reference:

H. Brendan McMahan et al., "Communication-efficient learning of deep networks from decentralized data", AISTATS'17 https://arxiv.org/abs/1602.05629