myanmar-ghost / federated /server_config.yaml
amkyawdev's picture
Add configs
12b35a0 verified
Raw
History Blame Contribute Delete
401 Bytes
# Federated Learning Server Configuration
server:
address: "[::]:8080"
num_rounds: 5
strategy:
name: fedavg
fraction_fit: 0.5
fraction_evaluate: 0.5
min_fit_clients: 2
min_evaluate_clients: 2
min_available_clients: 2
training:
local_epochs: 1
batch_size: 32
learning_rate: 1e-4
weight_decay: 0.01
model:
name: myanmar_ghost_federated
num_classes: 4
hidden_dim: 768