search

ki-ljl / FedProx-PyTorch

PyTorch implementation of FedProx (Federated Optimization for Heterogeneous Networks, MLSys 2020).
MIT License
99 stars 19 forks source link
federated-learning fedprox pytorch

readme

FedProx-PyTorch

PyTorch implementation of FedProx (Federated Optimization for Heterogeneous Networks, MLSys 2020).

Environment

numpy==1.18.5

pytorch==1.10.1+cu111

Usage

python main.py