-
Notifications
You must be signed in to change notification settings - Fork 2
/
config_example.json
46 lines (45 loc) · 1.17 KB
/
config_example.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
{
"model_name": "pholid_sample3_seed66",
"Input": {
"userroot": "/home/hexin/",
"train": "/Desktop/hexin/dataset/train.txt",
"valid": "none",
"test": "Desktop/hexin/dataset/test.txt",
"log": "/Desktop/hexin/dataset/log/",
"_comment": "Input your data dir here, each line: data_file_path lable_index segment_len"
},
"model_config": {
"model": "pholid",
"feat_dim": 1024,
"reduc_dim": 256,
"d_k": 64,
"d_ff": 2048,
"n_heads": 8,
"n_language": 10,
"_comment": "Model configurations, do not change unless you need to modify the model"
},
"optim_config": {
"learning_rate": 0.0001,
"epochs": 13,
"batch": 128,
"optimizer": "Adam",
"scheduler": "warmup_cosine",
"num_work": 4,
"device": 3,
"DDP": False,
"SSL_epochs":3,
"warmup_step": -1,
"valid_epochs": 3,
"seed": 66,
"ptc_dist":"euclidean",
"margin":0.3,
"weight_lid":0.95,
"weight_pht":0.15,
"weight_ssl":0.05,
"phn_ssl":"seg",
"nega_frames":3,
"hard_sample_epochs":0,
"_comment": "warmup_step = -1 denotes default value, num_work is better to be your_cpu_cores/4"
},
"kaldi": "/Desktop/kaldi/"
}