flower-server/pyproject.toml at master · IoTFedLab/flower-server · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
# =====================================================================
# For a full TOML configuration guide, check the Flower docs:
# https://flower.ai/docs/framework/how-to-configure-pyproject-toml.html
# =====================================================================

[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[project]
name = "iot-fed"
version = "1.0.3"
description = ""
license = "Apache-2.0"
# Dependencies for your Flower App
dependencies = [
    "flwr[simulation]>=1.23.0",
    "flwr-datasets[vision]>=0.5.0",
    "torch==2.7.1",
    "torchvision==0.22.1",
    "timm>=0.9.0",
    "pillow>=10.0.0",
]

[tool.hatch.build.targets.wheel]
packages = ["."]

[tool.flwr.app]
publisher = "minter"

# Point to your ServerApp and ClientApp objects
[tool.flwr.app.components]
serverapp = "iot_fed.server_app:app"
clientapp = "iot_fed.client_app:app"

# Custom config values accessible via `context.run_config`
[tool.flwr.app.config]
num-server-rounds = 5
fraction-train = 1.0
local-epochs = 4
lr = 0.00005
# FedProx configuration
fedprox-mu = 2.0  # Proximal term coefficient (0.0 = FedAvg, >0 = FedProx)
# Path to pretrained checkpoint (update this path to your actual checkpoint location)
checkpoint-path = "checkpoints/final_model.pt"
# Resume from previous federated learning result
resume-from-final = false  # true로 설정하면 final_model.pt에서 이어서 학습
# Dataset configuration
use-cifar10 = false  # Set to true to use CIFAR-10, false for custom skin disease dataset
data-root = "data/train"
val-data-root = "data/validation"

# Default federation to use when running the app
[tool.flwr.federations]
default = "remote-federation"  # 배포용

# Local simulation federation with virtual SuperNodes
[tool.flwr.federations.local-simulation]
options.num-supernodes = 3  # 민지, 시우(총 2명)

# Remote federation for 3 real devices (실제 배포용)
[tool.flwr.federations.remote-federation]
address = "127.0.0.1:9093"  # ServerApp이 로컬 SuperLink에 연결 (같은 서버에서 실행)
insecure = true
options.num-supernodes = 3  # 총 3대 디바이스 (이 수만큼 연결될 때까지 대기)