An implementation of Federated Learning(FL) by using Pytorch

Environment

torch >= 1.7.0
yaml
omegaconf
visdom
torchmetrics
Pillow 8.2.0
PrettyTable

How to use

you can directly run the code with the corrosponding dataset name, for example, the simulation can start with

python mnist.py

Copyright

• The code is provided by Hui Zeng from NUDT. If you have any question, please contact zenghui35@outlook.com.