├── README.md ├── api.py ├── config.py ├── core ├── fed_avg.py ├── fed_pd.py ├── fedavg_distill.py ├── ffgb_distill.py ├── imbalance_fl.py ├── ratio_loss_fl.py └── scaffold.py ├── model ├── convnet.py ├── mlp.py └── resnet.py ├── requirements.txt ├── run_FFL.py ├── run_FL.py ├── run_PD_FL.py ├── run_experiment.sh └── utils ├── data_utils.py ├── general_utils.py ├── logger_utils.py ├── loss_utils.py ├── model_utils.py └── test_utils.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenzebang/Federated-Learning-Pytorch/HEAD/README.md -------------------------------------------------------------------------------- /api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenzebang/Federated-Learning-Pytorch/HEAD/api.py -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenzebang/Federated-Learning-Pytorch/HEAD/config.py -------------------------------------------------------------------------------- /core/fed_avg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenzebang/Federated-Learning-Pytorch/HEAD/core/fed_avg.py -------------------------------------------------------------------------------- /core/fed_pd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenzebang/Federated-Learning-Pytorch/HEAD/core/fed_pd.py -------------------------------------------------------------------------------- /core/fedavg_distill.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenzebang/Federated-Learning-Pytorch/HEAD/core/fedavg_distill.py -------------------------------------------------------------------------------- /core/ffgb_distill.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenzebang/Federated-Learning-Pytorch/HEAD/core/ffgb_distill.py -------------------------------------------------------------------------------- /core/imbalance_fl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenzebang/Federated-Learning-Pytorch/HEAD/core/imbalance_fl.py -------------------------------------------------------------------------------- /core/ratio_loss_fl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenzebang/Federated-Learning-Pytorch/HEAD/core/ratio_loss_fl.py -------------------------------------------------------------------------------- /core/scaffold.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenzebang/Federated-Learning-Pytorch/HEAD/core/scaffold.py -------------------------------------------------------------------------------- /model/convnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenzebang/Federated-Learning-Pytorch/HEAD/model/convnet.py -------------------------------------------------------------------------------- /model/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenzebang/Federated-Learning-Pytorch/HEAD/model/mlp.py -------------------------------------------------------------------------------- /model/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenzebang/Federated-Learning-Pytorch/HEAD/model/resnet.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torch 2 | tqdm 3 | ray 4 | numpy 5 | torchvision -------------------------------------------------------------------------------- /run_FFL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenzebang/Federated-Learning-Pytorch/HEAD/run_FFL.py -------------------------------------------------------------------------------- /run_FL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenzebang/Federated-Learning-Pytorch/HEAD/run_FL.py -------------------------------------------------------------------------------- /run_PD_FL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenzebang/Federated-Learning-Pytorch/HEAD/run_PD_FL.py -------------------------------------------------------------------------------- /run_experiment.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenzebang/Federated-Learning-Pytorch/HEAD/run_experiment.sh -------------------------------------------------------------------------------- /utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenzebang/Federated-Learning-Pytorch/HEAD/utils/data_utils.py -------------------------------------------------------------------------------- /utils/general_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenzebang/Federated-Learning-Pytorch/HEAD/utils/general_utils.py -------------------------------------------------------------------------------- /utils/logger_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenzebang/Federated-Learning-Pytorch/HEAD/utils/logger_utils.py -------------------------------------------------------------------------------- /utils/loss_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenzebang/Federated-Learning-Pytorch/HEAD/utils/loss_utils.py -------------------------------------------------------------------------------- /utils/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenzebang/Federated-Learning-Pytorch/HEAD/utils/model_utils.py -------------------------------------------------------------------------------- /utils/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenzebang/Federated-Learning-Pytorch/HEAD/utils/test_utils.py --------------------------------------------------------------------------------