Spaces:
Sleeping
Sleeping
File size: 1,199 Bytes
3e25ded |
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 |
"""huggingface_example: A Flower / Hugging Face app."""
from flwr.common import Context, ndarrays_to_parameters
from flwr.server import ServerApp, ServerAppComponents, ServerConfig
from flwr.server.strategy import FedAvg
from .task import get_params, get_model
def server_fn(context: Context) -> ServerAppComponents:
"""Construct components for ServerApp."""
# Construct ServerConfig
num_rounds = context.run_config["num-server-rounds"]
config = ServerConfig(num_rounds=num_rounds)
# Set global model initialization
model_name = context.run_config["model-name"]
ndarrays = get_params(get_model(model_name))
global_model_init = ndarrays_to_parameters(ndarrays)
# Define strategy
fraction_fit = context.run_config["fraction-fit"]
fraction_evaluate = context.run_config["fraction-evaluate"]
strategy = FedAvg(
fraction_fit=fraction_fit,
fraction_evaluate=fraction_evaluate,
min_fit_clients=1, # Wait for at least one client to start
min_available_clients=1,
initial_parameters=global_model_init,
)
return ServerAppComponents(config=config, strategy=strategy)
app = ServerApp(server_fn=server_fn)
|