-
Notifications
You must be signed in to change notification settings - Fork 0
/
cddb_training.json
36 lines (36 loc) · 978 Bytes
/
cddb_training.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
{
"batch_size" : 2,
"batch_size_eval" : 512,
"epochs" : 20,
"warmup_epoch" : 1,
"init_lr" : 0.01,
"lrate" : 0.01,
"label_smoothing" : 0.1,
"topk_classes" : 5,
"K" : 7,
"enable_prev_prompt" : false,
"ensembling" : [false, false, true, false],
"precision" : "fp16",
"dataset": "cddb",
"task_name": ["gaugan", "biggan", "wild", "whichfaceisreal", "san"],
"data_path": "/home/francesco.laiti/datasets/CDDB/",
"multiclass": [0, 0, 0, 0, 0],
"class_order": [0, 1, 2, 3, 4, 5, 6, 7, 8, 9],
"memory_size": 0,
"memory_per_class": 0,
"fixed_memory": true,
"shuffle": false,
"init_cls": 2,
"increment": 2,
"embd_dim" : 768,
"torch_seed" : 2,
"seed": [1993],
"EPSILON" : 1e-8,
"init_milestones" : [20,30,40],
"init_lr_decay" : 0.1,
"init_weight_decay" : 0.0005,
"milestones" : [20, 30],
"lrate_decay" : 0.1,
"weight_decay" : 2e-4,
"num_workers" : 12
}