An implementation of the FedAvg algorithm

This repo is a basic PyTorch implementation of the Federated Learning FedAvg algorithm.

The code is adapted from GitHub-FL.

Run

It is easy to run with:

python fedavg.py --dataset mnist --iid 0 --model mlp1 --rounds 100

You can also check ./utils/arguments.py to personalize different parameters.

Example results