This repository implements the FedAvg algorithm on Pytorch.
- Python 3.6+
- PyTorch 1.0+
# Start training with:
python main.py
# You can manually modify the param.yaml file and then start training with:
python main.py --param param.yaml
Currently, the code only implements versions of ResNet18+Cifar10. In the future, training processes for other models and datasets may be added.