-
Notifications
You must be signed in to change notification settings - Fork 0
/
NIH_Experiment.json
113 lines (102 loc) · 2.5 KB
/
NIH_Experiment.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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
{
"DATASET": "NIH",
"PATH": "Datasets/NIH/",
"TARGET": "Airspace_Opacity",
"LABELER_IDS": [[4323195249, 4295232296], [4295349121, 4295342357], [4295342357, 4295354117]],
"K": 10,
"SEEDS": [1, 2, 3, 4],
"GT": true,
"MOD": ["confidence", "ssl", "disagreement_diff"],
"OVERLAP": [0, 100],
"SAMPLE_EQUAL": [false, true],
"SETTING": ["SSL_AL"],
"NUM_EXPERTS": 2,
"NUM_CLASSES": 2,
"EXPERT_PREDICT": ["target", "right"],
"AL": {
"INITIAL_SIZE": [4, 8, 16, 32],
"EPOCH_TRAIN": 40,
"n_dataset": 2,
"BATCH_SIZE": 4,
"BATCH_SIZE_VAL": 32,
"ROUNDS": [2, 4],
"LABELS_PER_ROUND": [4, 8, 16],
"EPOCHS_DEFER": 10,
"COST": [[0, 0]],
"PRELOAD": true,
"PREPROCESS": true,
"SSL_EPOCHS": 3
},
"SSL": {
"PREBUILD": false,
"TRAIN_BATCH_SIZE": 128,
"TRAIN_BATCH_SIZE": 254,
"TEST_BATCH_SIZE": 254,
"N_EPOCHS": 5,
"BATCHSIZE": 16,
"N_IMGS_PER_EPOCH": 4381
},
"L2D": {
"TRAIN_BATCH_SIZE": 128,
"TEST_BATCH_SIZE": 128,
"PRELOAD": true,
"PREBUILD": true,
"EPOCHS": 50,
"VERMA": {},
"HEMMER": {
"EPOCHS": 50,
"LR": 5e-3,
"USE_LR_SCHEDULER": false,
"DROPOUT": 0.00,
"NUM_HIDDEN_UNITS": 30
}
},
"NEPTUNE": {
"NEPTUNE": false
},
"EMBEDDED": {
"ARGS": {
"dataset": "nih",
"model": "resnet50",
"num_classes": 2,
"batch": 128,
"lr": 0.001
},
"EPOCHS": 30
},
"CIFAR100": {
"EXPERTS": {
"123": {
"strength": 60,
"binary": false,
"num_classes": 20,
"per_s": 1.0,
"per_w": 0.0
},
"456": {
"strength": 60,
"binary": false,
"num_classes": 20,
"per_s": 1.0,
"per_w": 0.0
}
}
},
"epochs_pretrain": [0],
"batch_size": 64,
"alpha": 1.0,
"epochs": 50,
"patience": 35,
"expert_type": "MLPMixer",
"n_classes": 2,
"k": 0,
"n_experts": 2,
"lr": 0.001,
"weight_decay": 5e-4,
"warmup_epochs": 5,
"loss_type": "ova",
"ckp_dir": "/Models",
"experiment_name": "multiple_experts",
"cluster": true,
"IMAGE_SIZE": 128
}