c-gabri / Federated-Learning-PyTorch

PyTorch implementation of Federated Learning algorithms FedSGD, FedAvg, FedAvgM, FedIR, FedVC, FedProx and standard SGD, applied to visual classification. Client distributions are synthesized with arbitrary non-identicalness and imbalance (Dirichlet priors). Client systems can be arbitrarily heterogeneous. Several mobile-friendly models are prov…
67Updated 2 years ago

Related projects

Alternatives and complementary repositories for Federated-Learning-PyTorch