Hi all!
I have updated my old FL simulation repo. (I made it for personal research purpose, FYI):
(https://github.com/vaseline555/Federated-Learning-PyTorch)
It now supports
- extensive datasets including all
torchvision.datasets, torchtext.datasets, LEAF benchmark
(now you DON'T have to prepare raw data manually! - I've ported ALL shell commands into python😁),
- various models (some are adapted from
transformers library),
- with some basic FL algorithms (
FedAvg, FedSGD, and FedProx),
- and popular non-IID simulation scenario.
If you have interest, please visit my repository.
Thank you and also welcome any feedbacks or PRs. 😀