FedBoosting/config.json at main · Rand2AI/FedBoosting · GitHub

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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
{
  "NAME": "Federated Learning with Homomorphic Encryption on CRNN",

  "AUTHOR": "Ren, Hanchi and Deng, Jingjing, Xie, Xianghua, Xiaoke Ma and Yichuan Wang",

  "WORKMODE": "Estimation",

  "MODE_SET": ["Estimation", "Inference"],

  "NETWORK":
  {
    "BACKBONE_CNN_FLOW": "Original",
    "BACKBONE_CNN_AVAILABLE": ["Original"],
    "BACKBONE_CNN_LAYER": 121,
    "BACKBONE_RNN_CELL": "LSTM",
    "BACKBONE_RNN1_UNIT": [256, 256],
    "BACKBONE_RNN2_UNIT": [256],
    "BACKBONE_RNN_AVAILABLE": ["LSTM", "GRU"],
    "DICTIONARY_PATH": "/home/hans/WorkSpace/Data/Text/Models/Transcriptor/DATA/character_upper.txt",
    "DICTIONARY_SIZE": 37,
    "MAX_LABEL_LENGTH": 32,
    "IMAGE_HEIGHT": 32,
    "IMAGE_WIDTH": 0,
    "IMAGE_CHANNEL" : 3
  },

  "DEVICE":
  {
    "DEVICE_TOUSE": "GPU",
    "DEVICE_GPUID": [0],
    "CLIENT_GPU_FRACTION": 0.80,
    "SERVER_GPU_FRACTION": 0.45
  },

  "ESTIMATION":
  {
    "DEBUG": false,
    "FL_STRATEGY": "FedAvg",
    "FL_STRAEGY_SET": ["FedAvg, FedBoost"],
    "FL_STRATEGY_METRICS": "loss",
    "FL_STRATEGY_METRICS_SET": ["acc", "loss"],
    "OPT_BACKPROP": "adadelta",
    "OPT_BACKPROP_SET": ["SGD", "ADAM", "rmsprop", "adadelta"],
    "LEARNING_RATE": 0.05,
    "SGD_MOMENTUM": 0.9,
    "SGD_DECAY": 0.0005,
    "SGD_NESTEROV": "True",
    "FINETUNE" : "NO",
    "WEIGHT_TOLOAD": "/home/hans/WorkSpace/Data/Text/Models/Transcriptor/FL_CRNN/FedBoost/initial-global_weights.h5",
    "STEMFREEZE": "NO",
    "TRN_RATIO": 0.9,
    "IMAGE_SIZE": [32, 100],
    "NUM_MAXEPOCH": 50,
    "NUM_MINEPOCH": 1,
    "BATCH_SIZE": 800,
    "ROUNDS": 100,
    "CLIENTS": 2,
    "CLIENT_EPOCH": [1,1],
    "TRAIN_NAME_LIST": ["SynthText90K/SynthText90K_20190911_train.json",
                        "SynthText80K/SynthText80K_20190923_cropped_train.json"],
    "TEST_NAME_LIST": ["IIIT.5K/IIIT5K_test_20190829.json",
                        "SVT/svt_test_20190828_cropped.json",
                        "ICDAR2015/ICDAR2015_test_20190829.json"],
    "TRAIN_ROOT_PATH": "/home/hans/WorkSpace/Data/Text/Models/Transcriptor/DATA/",
    "MODEL_ROOT_PATH": "/home/hans/WorkSpace/Data/Text/Models/Transcriptor/FL_CRNN/",
    "SMALL_SLICE": 100,
    "SMALL_SLICE_INFO": "smaller SMALL_SLICE, smaller accuracy reduction. must more than client number, but too large may cause error.",
    "DP": false
  },
  "SERVER":
  {
    "SERVER_DOMAIN": "csjdsvr.swan.ac.uk",
    "SERVER_AUTH_KEY": "hahaha",
    "SERVER_PORT": 7777
  },
  "CLIENT":
  {
    "CLIENT_PORT": 8888,
    "CLIENT_DOMAIN": "csjdsvr.swan.ac.uk",
    "CLIENT_AUTH_KEY": "heiheihei",
    "KEY_LENGTH": 128,
    "SAVE_CLIENT_MODEL": 1,
    "ROUNDS_SERVER_GET_MODEL": 1
  },
  "INFERENCE":
  {
    "SAVE_IMG": 0,
    "WEIGHT_TOLOAD": "FedBoost_HE/20200711-214522-1E800B_FedBoost_HE/global/Round_0023-global_weights-test_acc:76.07.h5"
  }
}