/FedProx-PyTorch

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

Primary LanguagePythonMIT LicenseMIT

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