FederatedAvg Leaning

1 Introduce

This repository records the algorithm for federated average.

The implementation is pytorch.

2 how to run

run the main.py.

3 how to change

if you want check the default data, you can change the getData.py. And if you want change the model in Federated Learning, you just need to change the net.py in the model folder. See main.py for communication rounds and other variables