-
Notifications
You must be signed in to change notification settings - Fork 2
/
config.py
75 lines (74 loc) · 2.04 KB
/
config.py
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
66
67
68
69
70
71
72
73
74
75
'''
@ Author: Yue
@ Date: 2022-06-21 08:17:03
@ LastEditTime: 2022-06-22 08:10:19
@ FilePath: /gsy/FedDUB/config.py
@ Description:
@ Email: 21S151140@stu.hit.edu.cn , Tel: +86-13184023012
@ Copyright (c) 2022 by HITSZ / Songyue Guo, All Rights Reserved.
'''
default_param_dicts = {
"famnist": {
"dataset": ["famnist"],
"split": ["dirichlet"],
"dir_alpha": [None],
"n_classes": [10],
"n_clients": [None],
"n_max_sam": [None],
"fed_ratio": [None],
"net": [None],
"max_round": [None],
"test_round": [None],
"local_epochs": [None],
"local_steps": [None],
"batch_size": [64],
"optimizer": ["SGD"],
"lr": [0.03],
"momentum": [0.9],
"weight_decay": [1e-5],
"max_grad_norm": [100.0],
"cuda": [True],
},
"cifar10": {
"dataset": ["cifar10"],
"split": ["dirichlet"],
"dir_alpha": [None],
"n_classes": [10],
"n_clients": [None],
"n_max_sam": [None],
"fed_ratio": [None],
"net": [None],
"max_round": [None],
"test_round": [None],
"local_epochs": [None],
"local_steps": [None],
"batch_size": [64],
"optimizer": ["SGD"],
"lr": [0.03],
"momentum": [0.9],
"weight_decay": [1e-5],
"max_grad_norm": [100.0],
"cuda": [True],
},
"cifar100": {
"dataset": ["cifar100"],
"split": ["dirichlet"],
"dir_alpha": [None],
"n_classes": [100],
"n_clients": [None],
"n_max_sam": [None],
"c_ratio": [None],
"net": [None],
"max_round": [None],
"test_round": [None],
"local_epochs": [None],
"local_steps": [None],
"batch_size": [64],
"optimizer": ["SGD"],
"lr": [0.03],
"momentum": [0.9],
"weight_decay": [1e-5],
"max_grad_norm": [100.0],
"cuda": [True],
},
}