-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathFedAGit-Nova-config.json
96 lines (96 loc) · 2.09 KB
/
FedAGit-Nova-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
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
91
92
93
94
95
96
{
"wandb": {
"enabled": false,
"project": "FedAGit论文",
"name": "FedAGit+nova"
},
"global": {
"mode": "thread",
"use_file_system": false,
"multi_gpu": true,
"experiment": "FedAGit/essay/1",
"stale": [0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,10,10,10,10,10,10,10,10,10,10,30,30,30,30,30],
"dataset": {
"path": "dataset.CIFAR10.CIFAR10",
"params": {}
},
"iid": {
"beta": 0.5
},
"client_num": 30
},
"server": {
"path": "server.NormalServer.NormalServer",
"epochs": 200,
"model": {
"path": "torchvision.models.resnet.resnet18",
"params": {
"num_classes": 10
}
},
"scheduler": {
"path": "scheduler.VDAScheduler.VDAScheduler",
"schedule":{
"path": "schedule.RandomSchedule.RandomSchedule",
"params": {
"c_ratio": 0.7
}
},
"time_window": 0
},
"updater": {
"path": "updater.SyncUpdater.SyncUpdater",
"update": {
"path": "update.FedNova.FedNova",
"params": {
}
},
"loss": "torch.nn.functional.cross_entropy",
"params": {
}
}
},
"queue_manager": {
"path": "queuemanager.VCMQueueManager.VCMQueueManager",
"vcm": {
"update": "direct"
},
"receiver": {
"path": "receiver.VCMReceiver.VCMReceiver",
"params": {
}
},
"checker": {
"path": "checker.AllChecker.AllChecker",
"params": {
}
}
},
"client_manager": {
"path": "clientmanager.NormalClientManager.NormalClientManager"
},
"client": {
"path": "client.NovaClient.NovaClient",
"epochs": 2,
"batch_size": 128,
"model": {
"path": "torchvision.models.resnet.resnet18",
"params": {
"num_classes": 10
}
},
"loss": "torch.nn.functional.cross_entropy",
"transform": {
"path": "transforms.Transforms.TransformForCIFARFactory.createTransform",
"params": {}
},
"mu": 0,
"optimizer": {
"path": "torch.optim.SGD",
"params": {
"lr": 0.01,
"weight_decay": 0
}
}
}
}