Simple Federated Learning

Install

pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121
pip install flwr==1.6.0 numpy matplotlib

Run

Automated

python run_demo.py

Manual

python server.py

python client.py --client_id 0

python client.py --client_id 1

python client.py --client_id 2

python visualize.py