├── .gitattributes ├── LICENSE ├── README.md ├── config.py ├── data ├── mnist │ ├── generate_equal.py │ └── generate_random_niid.py └── synthetic │ └── generate_synthetic.py ├── main.py ├── plot ├── example.mat ├── plot_E.py ├── plot_K.py ├── plot_example.py └── plot_schemes.py ├── result └── mnist_all_data_0_equal_niid │ ├── 2019-11-24T12-05-13_fedavg4_logistic__wn10_tn100_sd0_lr0.1_ep5_bs64_a │ ├── eval.event │ │ └── events.out.tfevents.1574568313.Kirschs-MacBook-Pro.local │ ├── metrics.json │ └── train.event │ │ └── events.out.tfevents.1574568313.Kirschs-MacBook-Pro.local │ └── 2019-11-24T13-46-13_fedavg4_logistic__wn10_tn100_sd0_lr0.1_ep5_bs64_a │ ├── eval.event │ └── events.out.tfevents.1574574373.Kirschs-MacBook-Pro.local │ └── train.event │ └── events.out.tfevents.1574574373.Kirschs-MacBook-Pro.local └── src ├── models ├── client.py ├── model.py ├── vgg.py └── worker.py ├── optimizers └── gd.py ├── trainers ├── base.py ├── fedavg.py ├── fedavg4.py ├── fedavg5.py └── fedavg9.py └── utils ├── flops_counter.py ├── tool_utils.py ├── torch_utils.py └── worker_utils.py /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lx10077/fedavgpy/HEAD/.gitattributes -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lx10077/fedavgpy/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lx10077/fedavgpy/HEAD/README.md -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lx10077/fedavgpy/HEAD/config.py -------------------------------------------------------------------------------- /data/mnist/generate_equal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lx10077/fedavgpy/HEAD/data/mnist/generate_equal.py -------------------------------------------------------------------------------- /data/mnist/generate_random_niid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lx10077/fedavgpy/HEAD/data/mnist/generate_random_niid.py -------------------------------------------------------------------------------- /data/synthetic/generate_synthetic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lx10077/fedavgpy/HEAD/data/synthetic/generate_synthetic.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lx10077/fedavgpy/HEAD/main.py -------------------------------------------------------------------------------- /plot/example.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lx10077/fedavgpy/HEAD/plot/example.mat -------------------------------------------------------------------------------- /plot/plot_E.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lx10077/fedavgpy/HEAD/plot/plot_E.py -------------------------------------------------------------------------------- /plot/plot_K.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lx10077/fedavgpy/HEAD/plot/plot_K.py -------------------------------------------------------------------------------- /plot/plot_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lx10077/fedavgpy/HEAD/plot/plot_example.py -------------------------------------------------------------------------------- /plot/plot_schemes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lx10077/fedavgpy/HEAD/plot/plot_schemes.py -------------------------------------------------------------------------------- /result/mnist_all_data_0_equal_niid/2019-11-24T12-05-13_fedavg4_logistic__wn10_tn100_sd0_lr0.1_ep5_bs64_a/eval.event/events.out.tfevents.1574568313.Kirschs-MacBook-Pro.local: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lx10077/fedavgpy/HEAD/result/mnist_all_data_0_equal_niid/2019-11-24T12-05-13_fedavg4_logistic__wn10_tn100_sd0_lr0.1_ep5_bs64_a/eval.event/events.out.tfevents.1574568313.Kirschs-MacBook-Pro.local -------------------------------------------------------------------------------- /result/mnist_all_data_0_equal_niid/2019-11-24T12-05-13_fedavg4_logistic__wn10_tn100_sd0_lr0.1_ep5_bs64_a/metrics.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lx10077/fedavgpy/HEAD/result/mnist_all_data_0_equal_niid/2019-11-24T12-05-13_fedavg4_logistic__wn10_tn100_sd0_lr0.1_ep5_bs64_a/metrics.json -------------------------------------------------------------------------------- /result/mnist_all_data_0_equal_niid/2019-11-24T12-05-13_fedavg4_logistic__wn10_tn100_sd0_lr0.1_ep5_bs64_a/train.event/events.out.tfevents.1574568313.Kirschs-MacBook-Pro.local: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lx10077/fedavgpy/HEAD/result/mnist_all_data_0_equal_niid/2019-11-24T12-05-13_fedavg4_logistic__wn10_tn100_sd0_lr0.1_ep5_bs64_a/train.event/events.out.tfevents.1574568313.Kirschs-MacBook-Pro.local -------------------------------------------------------------------------------- /result/mnist_all_data_0_equal_niid/2019-11-24T13-46-13_fedavg4_logistic__wn10_tn100_sd0_lr0.1_ep5_bs64_a/eval.event/events.out.tfevents.1574574373.Kirschs-MacBook-Pro.local: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lx10077/fedavgpy/HEAD/result/mnist_all_data_0_equal_niid/2019-11-24T13-46-13_fedavg4_logistic__wn10_tn100_sd0_lr0.1_ep5_bs64_a/eval.event/events.out.tfevents.1574574373.Kirschs-MacBook-Pro.local -------------------------------------------------------------------------------- /result/mnist_all_data_0_equal_niid/2019-11-24T13-46-13_fedavg4_logistic__wn10_tn100_sd0_lr0.1_ep5_bs64_a/train.event/events.out.tfevents.1574574373.Kirschs-MacBook-Pro.local: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lx10077/fedavgpy/HEAD/result/mnist_all_data_0_equal_niid/2019-11-24T13-46-13_fedavg4_logistic__wn10_tn100_sd0_lr0.1_ep5_bs64_a/train.event/events.out.tfevents.1574574373.Kirschs-MacBook-Pro.local -------------------------------------------------------------------------------- /src/models/client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lx10077/fedavgpy/HEAD/src/models/client.py -------------------------------------------------------------------------------- /src/models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lx10077/fedavgpy/HEAD/src/models/model.py -------------------------------------------------------------------------------- /src/models/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lx10077/fedavgpy/HEAD/src/models/vgg.py -------------------------------------------------------------------------------- /src/models/worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lx10077/fedavgpy/HEAD/src/models/worker.py -------------------------------------------------------------------------------- /src/optimizers/gd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lx10077/fedavgpy/HEAD/src/optimizers/gd.py -------------------------------------------------------------------------------- /src/trainers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lx10077/fedavgpy/HEAD/src/trainers/base.py -------------------------------------------------------------------------------- /src/trainers/fedavg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lx10077/fedavgpy/HEAD/src/trainers/fedavg.py -------------------------------------------------------------------------------- /src/trainers/fedavg4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lx10077/fedavgpy/HEAD/src/trainers/fedavg4.py -------------------------------------------------------------------------------- /src/trainers/fedavg5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lx10077/fedavgpy/HEAD/src/trainers/fedavg5.py -------------------------------------------------------------------------------- /src/trainers/fedavg9.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lx10077/fedavgpy/HEAD/src/trainers/fedavg9.py -------------------------------------------------------------------------------- /src/utils/flops_counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lx10077/fedavgpy/HEAD/src/utils/flops_counter.py -------------------------------------------------------------------------------- /src/utils/tool_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lx10077/fedavgpy/HEAD/src/utils/tool_utils.py -------------------------------------------------------------------------------- /src/utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lx10077/fedavgpy/HEAD/src/utils/torch_utils.py -------------------------------------------------------------------------------- /src/utils/worker_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lx10077/fedavgpy/HEAD/src/utils/worker_utils.py --------------------------------------------------------------------------------