Text-to-Speech
TensorBoard
Ukrainian
Yehor commited on
Commit
f3402b8
·
verified ·
1 Parent(s): 80718cd

Upload 3 files

Browse files
radtts-pp-bgap-model/config.json ADDED
@@ -0,0 +1,224 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "train_config": {
3
+ "output_directory": "outdir_pp_bgap_model",
4
+ "epochs": 10000000,
5
+ "optim_algo": "RAdam",
6
+ "learning_rate": 0.0001,
7
+ "weight_decay": 1e-06,
8
+ "sigma": 1.0,
9
+ "iters_per_checkpoint": 1000,
10
+ "batch_size": 18,
11
+ "seed": null,
12
+ "checkpoint_path": "",
13
+ "ignore_layers": [],
14
+ "ignore_layers_warmstart": [],
15
+ "finetune_layers": [],
16
+ "include_layers": [],
17
+ "vocoder_config_path": "models/hifigan_22khz_config.json",
18
+ "vocoder_checkpoint_path": "models/hifigan_ljs_generator_v1.pt",
19
+ "log_attribute_samples": true,
20
+ "log_decoder_samples": true,
21
+ "warmstart_checkpoint_path": "models/model_radtss_pp_decoder_100000",
22
+ "use_amp": false,
23
+ "grad_clip_val": 1.0,
24
+ "loss_weights": {
25
+ "blank_logprob": -1,
26
+ "ctc_loss_weight": 0.1,
27
+ "binarization_loss_weight": 1.0,
28
+ "dur_loss_weight": 1.0,
29
+ "f0_loss_weight": 1.0,
30
+ "energy_loss_weight": 1.0,
31
+ "vpred_loss_weight": 1.0
32
+ },
33
+ "binarization_start_iter": 0,
34
+ "kl_loss_start_iter": 0,
35
+ "unfreeze_modules": "durf0energyvpred"
36
+ },
37
+ "data_config": {
38
+ "training_files": {
39
+ "LJS": {
40
+ "basedir": "filelists/",
41
+ "audiodir": "wavs",
42
+ "filelist": "3speakers_ukrainian_train_filelist.txt",
43
+ "lmdbpath": ""
44
+ }
45
+ },
46
+ "validation_files": {
47
+ "LJS": {
48
+ "basedir": "filelists/",
49
+ "audiodir": "wavs",
50
+ "filelist": "3speakers_ukrainian_val_filelist.txt",
51
+ "lmdbpath": ""
52
+ }
53
+ },
54
+ "dur_min": 0.1,
55
+ "dur_max": 10.2,
56
+ "sampling_rate": 22050,
57
+ "filter_length": 1024,
58
+ "hop_length": 256,
59
+ "win_length": 1024,
60
+ "n_mel_channels": 80,
61
+ "mel_fmin": 0.0,
62
+ "mel_fmax": 8000.0,
63
+ "f0_min": 80.0,
64
+ "f0_max": 640.0,
65
+ "max_wav_value": 32768.0,
66
+ "use_f0": true,
67
+ "use_log_f0": 0,
68
+ "use_energy_avg": true,
69
+ "use_scaled_energy": true,
70
+ "symbol_set": "ukrainian",
71
+ "cleaner_names": [
72
+ "ukrainian_cleaners"
73
+ ],
74
+ "heteronyms_path": "tts_text_processing/heteronyms",
75
+ "phoneme_dict_path": "tts_text_processing/cmudict-0.7b",
76
+ "p_phoneme": 0.0,
77
+ "handle_phoneme": "word",
78
+ "handle_phoneme_ambiguous": "ignore",
79
+ "include_speakers": null,
80
+ "n_frames": -1,
81
+ "betabinom_cache_path": "/home/yehor/RADTTS-Multiple-Voices/radtts/data_cache",
82
+ "lmdb_cache_path": "",
83
+ "use_attn_prior_masking": true,
84
+ "prepend_space_to_text": true,
85
+ "append_space_to_text": true,
86
+ "add_bos_eos_to_text": false,
87
+ "betabinom_scaling_factor": 1.0,
88
+ "distance_tx_unvoiced": true,
89
+ "mel_noise_scale": 0.0
90
+ },
91
+ "dist_config": {
92
+ "dist_backend": "nccl",
93
+ "dist_url": "tcp://localhost:54321"
94
+ },
95
+ "model_config": {
96
+ "n_speakers": 3,
97
+ "n_speaker_dim": 16,
98
+ "n_text": 185,
99
+ "n_text_dim": 512,
100
+ "n_flows": 8,
101
+ "n_conv_layers_per_step": 4,
102
+ "n_mel_channels": 80,
103
+ "n_hidden": 1024,
104
+ "mel_encoder_n_hidden": 512,
105
+ "dummy_speaker_embedding": false,
106
+ "n_early_size": 2,
107
+ "n_early_every": 2,
108
+ "n_group_size": 2,
109
+ "affine_model": "wavenet",
110
+ "include_modules": "decatndpmvpredapm",
111
+ "scaling_fn": "tanh",
112
+ "matrix_decomposition": "LUS",
113
+ "learn_alignments": true,
114
+ "use_speaker_emb_for_alignment": false,
115
+ "attn_straight_through_estimator": true,
116
+ "use_context_lstm": true,
117
+ "context_lstm_norm": "spectral",
118
+ "context_lstm_w_f0_and_energy": true,
119
+ "text_encoder_lstm_norm": "spectral",
120
+ "n_f0_dims": 1,
121
+ "n_energy_avg_dims": 1,
122
+ "use_first_order_features": true,
123
+ "unvoiced_bias_activation": "relu",
124
+ "decoder_use_partial_padding": true,
125
+ "decoder_use_unvoiced_bias": true,
126
+ "ap_pred_log_f0": true,
127
+ "ap_use_unvoiced_bias": false,
128
+ "ap_use_voiced_embeddings": true,
129
+ "dur_model_config": {
130
+ "name": "dap",
131
+ "hparams": {
132
+ "n_speaker_dim": 16,
133
+ "bottleneck_hparams": {
134
+ "in_dim": 512,
135
+ "reduction_factor": 16,
136
+ "norm": "weightnorm",
137
+ "non_linearity": "relu"
138
+ },
139
+ "take_log_of_input": true,
140
+ "arch_hparams": {
141
+ "out_dim": 1,
142
+ "n_layers": 2,
143
+ "n_channels": 256,
144
+ "kernel_size": 3,
145
+ "p_dropout": 0.25,
146
+ "in_dim": 48
147
+ }
148
+ }
149
+ },
150
+ "f0_model_config": {
151
+ "name": "bgap",
152
+ "hparams": {
153
+ "n_in_dim": 2,
154
+ "take_log_of_input": false,
155
+ "n_speaker_dim": 16,
156
+ "n_flows": 6,
157
+ "n_group_size": 2,
158
+ "n_layers": 4,
159
+ "kernel_size": 5,
160
+ "scaling_fn": "tanh",
161
+ "with_dilation": true,
162
+ "bottleneck_hparams": {
163
+ "in_dim": 512,
164
+ "reduction_factor": 16,
165
+ "norm": "weightnorm",
166
+ "non_linearity": "leakyrelu",
167
+ "use_partial_padding": true,
168
+ "kernel_size": 1
169
+ },
170
+ "n_bins": 16,
171
+ "use_quadratic": true,
172
+ "n_spline_steps": 4
173
+ }
174
+ },
175
+ "energy_model_config": {
176
+ "name": "bgap",
177
+ "hparams": {
178
+ "n_in_dim": 2,
179
+ "take_log_of_input": false,
180
+ "n_speaker_dim": 16,
181
+ "n_flows": 6,
182
+ "n_group_size": 4,
183
+ "n_layers": 4,
184
+ "kernel_size": 5,
185
+ "scaling_fn": "tanh",
186
+ "with_dilation": true,
187
+ "bottleneck_hparams": {
188
+ "in_dim": 512,
189
+ "reduction_factor": 16,
190
+ "norm": "weightnorm",
191
+ "non_linearity": "leakyrelu",
192
+ "use_partial_padding": true,
193
+ "kernel_size": 1
194
+ },
195
+ "n_bins": 16,
196
+ "use_quadratic": true,
197
+ "n_spline_steps": 4
198
+ }
199
+ },
200
+ "v_model_config": {
201
+ "name": "dap",
202
+ "hparams": {
203
+ "n_speaker_dim": 16,
204
+ "take_log_of_input": false,
205
+ "bottleneck_hparams": {
206
+ "in_dim": 512,
207
+ "reduction_factor": 16,
208
+ "norm": "weightnorm",
209
+ "non_linearity": "relu"
210
+ },
211
+ "arch_hparams": {
212
+ "out_dim": 1,
213
+ "n_layers": 2,
214
+ "n_channels": 256,
215
+ "kernel_size": 3,
216
+ "p_dropout": 0.5,
217
+ "lstm_type": "",
218
+ "use_linear": 1,
219
+ "in_dim": 48
220
+ }
221
+ }
222
+ }
223
+ }
224
+ }
radtts-pp-bgap-model/logs/events.out.tfevents.1683933666.desktop.2175284.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a3bcca9dd1e8b06e07cd145b57f67315d1af1c366b6983c2a8037ae31c0a513c
3
+ size 42834375
radtts-pp-bgap-model/model_bgap_50000.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f2dc49ada6e8169fe62f4eae7b3a7f7809763be33835edf83abc92e6066fab5e
3
+ size 1957493741