├── .gitignore ├── LICENSE ├── README.md ├── conf ├── base.yaml ├── base_kd.yaml ├── base_v2.yaml ├── client │ ├── cpu_client.yaml │ ├── gpu_client.yaml │ └── kd_client.yaml ├── model │ ├── resnet18.yaml │ └── simple_net.yaml └── strategy │ ├── custom_strategy.yaml │ ├── fedavg.yaml │ ├── strategy_kd.yaml │ └── strategy_model_saving.yaml ├── main.py ├── requirements.txt └── src ├── client.py ├── common.py ├── datasets.py ├── model_utils.py ├── models.py ├── server.py └── strategy.py /.gitignore: -------------------------------------------------------------------------------- 1 | *__pycache__/ 2 | data/ 3 | outputs/ 4 | *.DS_Store 5 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jafermarq/FlowerMonthly/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jafermarq/FlowerMonthly/HEAD/README.md -------------------------------------------------------------------------------- /conf/base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jafermarq/FlowerMonthly/HEAD/conf/base.yaml -------------------------------------------------------------------------------- /conf/base_kd.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jafermarq/FlowerMonthly/HEAD/conf/base_kd.yaml -------------------------------------------------------------------------------- /conf/base_v2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jafermarq/FlowerMonthly/HEAD/conf/base_v2.yaml -------------------------------------------------------------------------------- /conf/client/cpu_client.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jafermarq/FlowerMonthly/HEAD/conf/client/cpu_client.yaml -------------------------------------------------------------------------------- /conf/client/gpu_client.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jafermarq/FlowerMonthly/HEAD/conf/client/gpu_client.yaml -------------------------------------------------------------------------------- /conf/client/kd_client.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jafermarq/FlowerMonthly/HEAD/conf/client/kd_client.yaml -------------------------------------------------------------------------------- /conf/model/resnet18.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jafermarq/FlowerMonthly/HEAD/conf/model/resnet18.yaml -------------------------------------------------------------------------------- /conf/model/simple_net.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jafermarq/FlowerMonthly/HEAD/conf/model/simple_net.yaml -------------------------------------------------------------------------------- /conf/strategy/custom_strategy.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jafermarq/FlowerMonthly/HEAD/conf/strategy/custom_strategy.yaml -------------------------------------------------------------------------------- /conf/strategy/fedavg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jafermarq/FlowerMonthly/HEAD/conf/strategy/fedavg.yaml -------------------------------------------------------------------------------- /conf/strategy/strategy_kd.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jafermarq/FlowerMonthly/HEAD/conf/strategy/strategy_kd.yaml -------------------------------------------------------------------------------- /conf/strategy/strategy_model_saving.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jafermarq/FlowerMonthly/HEAD/conf/strategy/strategy_model_saving.yaml -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jafermarq/FlowerMonthly/HEAD/main.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | flwr[simulation]==1.7.0 2 | hydra-core==1.3.2 3 | tqdm==4.65.0 -------------------------------------------------------------------------------- /src/client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jafermarq/FlowerMonthly/HEAD/src/client.py -------------------------------------------------------------------------------- /src/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jafermarq/FlowerMonthly/HEAD/src/common.py -------------------------------------------------------------------------------- /src/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jafermarq/FlowerMonthly/HEAD/src/datasets.py -------------------------------------------------------------------------------- /src/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jafermarq/FlowerMonthly/HEAD/src/model_utils.py -------------------------------------------------------------------------------- /src/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jafermarq/FlowerMonthly/HEAD/src/models.py -------------------------------------------------------------------------------- /src/server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jafermarq/FlowerMonthly/HEAD/src/server.py -------------------------------------------------------------------------------- /src/strategy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jafermarq/FlowerMonthly/HEAD/src/strategy.py --------------------------------------------------------------------------------