ongzh / ScaffoldFL

Implementation of Scaffold and Fedprox for Federated Learning using PyTorch

Federated Learning (Scaffold and Fedprox) using PyTorch

Basic Implementation of:
Fedprox: Federated Optimization in Heterogeneous Networks
Scaffold: SCAFFOLD: Stochastic Controlled Averaging for Federated Learning

Models: VGG, CNN
DataSet: Cifar10

---

Note

Non-IID implementation and dataset used differ from the Scaffold Paper.
Sections of code inspired from FedAvg implementation by https://github.com/AshwinRJ/Federated-Learning-PyTorch