AshwinRJ/Federated-Learning-PyTorch
A PyTorch implementation of federated learning for distributed training of deep neural networks on decentralized data.

Velocity · 7d
+0.5
★ / day
Trend
→steady
star history
This repository implements the seminal federated learning paper for training deep networks from decentralized data. It provides both baseline and federated training approaches on standard datasets (MNIST, Fashion MNIST, CIFAR10) using simple MLP and CNN models. The implementation supports both IID and non-IID data distributions, allowing users to simulate realistic federated scenarios where client data may be heterogeneous.