Federated-Learning/config.json at main · BalajiAI/Federated-Learning · 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
{

     "data_config":{
                      "dataset_path": "./Data/CIFAR10",
                      "dataset_name": "CIFAR10",
                      "non_iid_per": 0
                   },


     "fed_config":{  "algorithm": "FedAvg",
                     "fraction_clients": 0.1,
                     "num_clients": 100,
                     "num_rounds": 100,
                     "num_epochs": 10,
                     "batch_size": 32,
                     "global_stepsize": 1,
                     "local_stepsize": 0.05,
                     "criterion": "torch.nn.CrossEntropyLoss"
                  },


     "model_config":{
                       "name": "CNN_Cifar"
                    },


     "global_config":{
                        "seed": 42,
                        "device": "cpu"
                     }

}