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