forked from TL-System/plato
-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.yml
60 lines (44 loc) · 1.08 KB
/
config.yml
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
clients:
# Type
type: simple
# The total number of clients
total_clients: 30
# The number of clients selected in each round
per_round: 30
# Should the clients compute test accuracy locally?
do_test: true
server:
address: 127.0.0.1
port: 8000
data:
# The training and testing dataset
datasource: MNIST
# Number of samples in each partition
partition_size: 2000
# IID or non-IID?
sampler: iid
# The random seed for sampling data
random_seed: 1
trainer:
# The type of the trainer
type: basic
# The maximum number of training rounds
rounds: 5
# The maximum number of clients running concurrently
max_concurrency: 4
# The target accuracy
target_accuracy: 0.94
# Number of epoches for local training in each communication round
epochs: 5
batch_size: 32
optimizer: SGD
# The machine learning model
model_name: lenet5
algorithm:
# Aggregation algorithm
type: fedavg
parameters:
optimizer:
lr: 0.01
momentum: 0.9
weight_decay: 0.0