/RobustFL-SOTA

A PyTorch Implementation of Federated Learning http://doi.org/10.5281/zenodo.4321561

Primary LanguagePython

Robust Federated Learning SOTA

This is the SOTA work:

Trimmed Mean

Median

FedAvg

Requirements

python>=3.6
pytorch>=0.4

Run

Federated learning with MLP and CNN is produced by:

python main_sota.py

See the arguments in options.py.