-
Notifications
You must be signed in to change notification settings - Fork 12
/
config.json
57 lines (56 loc) · 1.23 KB
/
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
{
"train": {
"train_batch_size":32,
"gradient_accumulate_every": 1,
"train_lr": 0.0001,
"train_num_steps": 1000000,
"ema_update_every": 10,
"ema_decay": 0.995,
"adam_betas": [0.9, 0.99],
"save_and_sample_every":1000,
"timesteps":1000,
"sampling_timesteps":1000,
"results_folder": "results",
"logs_folder" : "logs/vc",
"num_workers": 32,
"eps": 1e-09,
"keep_ckpts": 3,
"all_in_mem": false
},
"data": {
"training_files": "../vc_dataset_processed",
"val_files": "../val_dataset_processed",
"sampling_rate": 24000,
"hop_length": 256
},
"phoneme_encoder":{
"in_channels":256,
"hidden_channels":256,
"out_channels":256,
"n_layers":6,
"p_dropout":0.2
},
"f0_predictor":
{
"in_channels":256,
"hidden_channels":256,
"out_channels":1,
"attention_layers":10,
"n_heads":8,
"p_dropout":0.5
},
"prompt_encoder":{
"in_channels":100,
"hidden_channels":256,
"out_channels":256,
"n_layers":6,
"p_dropout":0.2
},
"diffusion_encoder":{
"in_channels":100,
"out_channels":100,
"hidden_channels":256,
"n_heads":8,
"p_dropout":0.2
}
}