-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.json
51 lines (49 loc) · 1.2 KB
/
config.json
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
{
"NAME": "Gradients Stand-in for Defending Deep Leakage in Federated Learning",
"AUTHOR": "YI HU, HANCHI REN, JINGJING DENG, YIMING LI, CHEN HU and XIANGHUA XIE",
"METHODS": "AdaDefense",
"METHODS_SET": ["Normal", "FedAvg", "FedKL", "EWWA", "AdaDefense"],
"DEBUG": true,
"DEVICE": {
"DEVICE_TOUSE": "GPU",
"DEVICE_GPUID": [0]
},
"NETWORK": {
"BACKBONE": "lenet",
"BACKBONE_SET": ["lenet", "resnet", "vgg16"],
"LAYER_NUMBER": 0,
"RESNET_LAYER_NUMBER_SET": [18, 34, 20, 32]
},
"DATA": {
"DATA_ROOT": "/home/hans/WorkSpace/Data/",
"TRAIN_DATA": "mnist",
"DATA_LIST": ["mnist", "cifar100", "cifar10"],
"IMG_SIZE": [32, 32],
"IS_IID": true
},
"OPTIMIZER": {
"OPT_BACKPROP": "sgd",
"OPT_BACKPROP_SET": ["sgd", "adam", "rmsprop", "adadelta"],
"LEARNING_RATE": 0.01,
"DECAY": 0.0001,
"MOMENTUM": 0.9,
"SGD_NESTEROV": true
},
"TRAIN": {
"FINETUNE": false,
"WEIGHT_TOLOAD": "",
"BATCH_SIZE": 64,
"ROUNDS": 200,
"SAVE_ROOT": "/home/hans/WorkSpace/Models/"
},
"FED": {
"OPTIMIZER": "adam",
"CLIENTS_NUM": 3,
"FRACTION": 1,
"CLIENT_EPOCH": 1
},
"TEST": {
"ROOT_PATH": "",
"MODEL_NAME": ""
}
}