sleepysss / FedProx-PyTorch

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

About

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

License:MIT License


Languages

Language:Python 100.0%