FedALG

The implementation of some federated learning algorithms based on PyTorch respectively.

Environment

conda create -n FL python=3.7

conda activate FL

conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=11.3 -c pytorch -c conda-forge
(modify pytorch and cuda version according to your device)

pip install -r requirements.txt

Datasets

1. MINST
2. EMINST
3. Cifar10

Usage

运行方式: Run the code

python main.py -config_path ./configs/fedavg.json