Thorsten-Voice commited on
Commit
0bc5779
1 Parent(s): b4f2ae4

TTS model trained on Coqui TTS (Tacotron2 DDC).

Browse files
scale_stats.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:13474283738326a4c01132b4c347e35910f08cdff8b1c595b52fc238c94bdb4d
3
+ size 10791
tts-config.json ADDED
@@ -0,0 +1,217 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "output_path": "/ssd/___tts/thorsten-ddc/output",
3
+ "logger_uri": null,
4
+ "run_name": "Thorsten-Dec2021-22k-DDC",
5
+ "project_name": null,
6
+ "run_description": "\ud83d\udc38Coqui trainer run.",
7
+ "print_step": 25,
8
+ "plot_step": 100,
9
+ "model_param_stats": false,
10
+ "wandb_entity": null,
11
+ "dashboard_logger": "tensorboard",
12
+ "log_model_step": null,
13
+ "save_step": 2000,
14
+ "save_n_checkpoints": 5,
15
+ "save_checkpoints": true,
16
+ "save_all_best": true,
17
+ "save_best_after": 10000,
18
+ "target_loss": null,
19
+ "print_eval": true,
20
+ "test_delay_epochs": -1,
21
+ "run_eval": true,
22
+ "run_eval_steps": null,
23
+ "distributed_backend": "nccl",
24
+ "distributed_url": "tcp://localhost:54321",
25
+ "mixed_precision": false,
26
+ "epochs": 1000,
27
+ "batch_size": 64,
28
+ "eval_batch_size": 16,
29
+ "grad_clip": 5.0,
30
+ "scheduler_after_epoch": false,
31
+ "lr": 0.0001,
32
+ "optimizer": "RAdam",
33
+ "optimizer_params": {
34
+ "betas": [
35
+ 0.9,
36
+ 0.998
37
+ ],
38
+ "weight_decay": 1e-06
39
+ },
40
+ "lr_scheduler": "NoamLR",
41
+ "lr_scheduler_params": {
42
+ "warmup_steps": 4000
43
+ },
44
+ "use_grad_scaler": false,
45
+ "cudnn_enable": true,
46
+ "cudnn_deterministic": false,
47
+ "cudnn_benchmark": false,
48
+ "training_seed": 54321,
49
+ "model": "tacotron2",
50
+ "num_loader_workers": 4,
51
+ "num_eval_loader_workers": 4,
52
+ "use_noise_augment": false,
53
+ "audio": {
54
+ "fft_size": 1024,
55
+ "win_length": 1024,
56
+ "hop_length": 256,
57
+ "frame_shift_ms": null,
58
+ "frame_length_ms": null,
59
+ "stft_pad_mode": "reflect",
60
+ "sample_rate": 22050,
61
+ "resample": false,
62
+ "preemphasis": 0.0,
63
+ "ref_level_db": 20,
64
+ "do_sound_norm": false,
65
+ "log_func": "np.log",
66
+ "do_trim_silence": true,
67
+ "trim_db": 60,
68
+ "do_rms_norm": false,
69
+ "db_level": null,
70
+ "power": 1.5,
71
+ "griffin_lim_iters": 60,
72
+ "num_mels": 80,
73
+ "mel_fmin": 50.0,
74
+ "mel_fmax": null,
75
+ "spec_gain": 1.0,
76
+ "do_amp_to_db_linear": true,
77
+ "do_amp_to_db_mel": true,
78
+ "pitch_fmax": 640.0,
79
+ "pitch_fmin": 0.0,
80
+ "signal_norm": false,
81
+ "min_level_db": -100,
82
+ "symmetric_norm": true,
83
+ "max_norm": 4.0,
84
+ "clip_norm": true,
85
+ "stats_path": null
86
+ },
87
+ "use_phonemes": true,
88
+ "phonemizer": "espeak",
89
+ "phoneme_language": "de",
90
+ "compute_input_seq_cache": false,
91
+ "text_cleaner": "basic_german_cleaners",
92
+ "enable_eos_bos_chars": false,
93
+ "test_sentences_file": "",
94
+ "phoneme_cache_path": "/ssd/___tts/thorsten-ddc/output/phoneme_cache",
95
+ "characters": {
96
+ "characters_class": "TTS.tts.utils.text.characters.IPAPhonemes",
97
+ "vocab_dict": null,
98
+ "pad": "<PAD>",
99
+ "eos": "<EOS>",
100
+ "bos": "<BOS>",
101
+ "blank": "<BLNK>",
102
+ "characters": "iy\u0268\u0289\u026fu\u026a\u028f\u028ae\u00f8\u0258\u0259\u0275\u0264o\u025b\u0153\u025c\u025e\u028c\u0254\u00e6\u0250a\u0276\u0251\u0252\u1d7b\u0298\u0253\u01c0\u0257\u01c3\u0284\u01c2\u0260\u01c1\u029bpbtd\u0288\u0256c\u025fk\u0261q\u0262\u0294\u0274\u014b\u0272\u0273n\u0271m\u0299r\u0280\u2c71\u027e\u027d\u0278\u03b2fv\u03b8\u00f0sz\u0283\u0292\u0282\u0290\u00e7\u029dx\u0263\u03c7\u0281\u0127\u0295h\u0266\u026c\u026e\u028b\u0279\u027bj\u0270l\u026d\u028e\u029f\u02c8\u02cc\u02d0\u02d1\u028dw\u0265\u029c\u02a2\u02a1\u0255\u0291\u027a\u0267\u02b2\u025a\u02de\u026b",
103
+ "punctuations": "!'(),-.:;? ",
104
+ "phonemes": null,
105
+ "is_unique": false,
106
+ "is_sorted": true
107
+ },
108
+ "add_blank": false,
109
+ "batch_group_size": 0,
110
+ "loss_masking": true,
111
+ "sort_by_audio_len": false,
112
+ "min_audio_len": 1,
113
+ "max_audio_len": Infinity,
114
+ "min_text_len": 1,
115
+ "max_text_len": Infinity,
116
+ "compute_f0": false,
117
+ "compute_linear_spec": false,
118
+ "precompute_num_workers": 8,
119
+ "start_by_longest": false,
120
+ "datasets": [
121
+ {
122
+ "name": "ljspeech",
123
+ "path": "/ssd/___tts/datasets/Thorsten-Neutral-Dec2021-22k/",
124
+ "meta_file_train": "metadata_train.csv",
125
+ "ignored_speakers": null,
126
+ "language": "",
127
+ "meta_file_val": "metadata_test.csv",
128
+ "meta_file_attn_mask": ""
129
+ }
130
+ ],
131
+ "test_sentences": [
132
+ "und \u00fcberzeugen dank feingef\u00fchl f\u00fcr den ganz gro\u00dfen leinwand-stoff.",
133
+ "zur schadensh\u00f6he gab es keine angaben.",
134
+ "au\u00dferdem k\u00f6nnen glasscheiben, w\u00e4nde und andere hindernisse das ergebnis beeinflussen.",
135
+ "ihre lippen m\u00fcssen dennoch nicht zwingend auf farbe verzichten.",
136
+ "es dauert lange, eine eigene Stimme zu entwickeln, aber jetzt wo ich sie habe, bin ich nie wieder still.",
137
+ "heute scheint die Sonne, Regen ist nicht zu erwarten.",
138
+ "die aktuelle Au\u00dfentemperatur betr\u00e4gt zw\u00f6lf Grad Celsius bei einer Luftfeuchtigkeit von achtunddrei\u00dfig Prozent."
139
+ ],
140
+ "eval_split_max_size": null,
141
+ "eval_split_size": 0.01,
142
+ "use_speaker_weighted_sampler": false,
143
+ "speaker_weighted_sampler_alpha": 1.0,
144
+ "use_language_weighted_sampler": false,
145
+ "language_weighted_sampler_alpha": 1.0,
146
+ "use_length_weighted_sampler": false,
147
+ "length_weighted_sampler_alpha": 1.0,
148
+ "use_gst": false,
149
+ "gst": null,
150
+ "gst_style_input": null,
151
+ "use_capacitron_vae": false,
152
+ "capacitron_vae": null,
153
+ "num_speakers": 1,
154
+ "num_chars": 131,
155
+ "r": 6,
156
+ "gradual_training": [
157
+ [
158
+ 0,
159
+ 6,
160
+ 64
161
+ ],
162
+ [
163
+ 10000,
164
+ 4,
165
+ 32
166
+ ],
167
+ [
168
+ 50000,
169
+ 3,
170
+ 32
171
+ ],
172
+ [
173
+ 100000,
174
+ 2,
175
+ 32
176
+ ]
177
+ ],
178
+ "memory_size": -1,
179
+ "prenet_type": "original",
180
+ "prenet_dropout": true,
181
+ "prenet_dropout_at_inference": false,
182
+ "stopnet": true,
183
+ "separate_stopnet": true,
184
+ "stopnet_pos_weight": 10.0,
185
+ "max_decoder_steps": 500,
186
+ "encoder_in_features": 512,
187
+ "decoder_in_features": 512,
188
+ "decoder_output_dim": 80,
189
+ "out_channels": 80,
190
+ "attention_type": "original",
191
+ "attention_heads": null,
192
+ "attention_norm": "sigmoid",
193
+ "attention_win": false,
194
+ "windowing": false,
195
+ "use_forward_attn": false,
196
+ "forward_attn_mask": false,
197
+ "transition_agent": false,
198
+ "location_attn": true,
199
+ "bidirectional_decoder": false,
200
+ "double_decoder_consistency": true,
201
+ "ddc_r": 6,
202
+ "speakers_file": null,
203
+ "use_speaker_embedding": false,
204
+ "speaker_embedding_dim": 512,
205
+ "use_d_vector_file": false,
206
+ "d_vector_file": false,
207
+ "d_vector_dim": null,
208
+ "seq_len_norm": false,
209
+ "decoder_loss_alpha": 0.25,
210
+ "postnet_loss_alpha": 0.25,
211
+ "postnet_diff_spec_alpha": 0.25,
212
+ "decoder_diff_spec_alpha": 0.25,
213
+ "decoder_ssim_alpha": 0.25,
214
+ "postnet_ssim_alpha": 0.25,
215
+ "ga_alpha": 5.0,
216
+ "github_branch": "* main"
217
+ }
tts-model_file.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:df1ce9b284491421e2054509beecb74d1bae53c67d1befeed5b2bbd990e89bb8
3
+ size 572143679
vocoder-config.json ADDED
@@ -0,0 +1,189 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "output_path": "/home/ubuntu/output-hifigan",
3
+ "logger_uri": null,
4
+ "run_name": "Thorsten-Neutral-Dec2021-22k-HifiGAN",
5
+ "project_name": null,
6
+ "run_description": "HifiGAN vocoder model for Thorsten Neutral Dec2021 22k Samplerate Tacotron2 DDC model",
7
+ "print_step": 25,
8
+ "plot_step": 100,
9
+ "model_param_stats": false,
10
+ "wandb_entity": null,
11
+ "dashboard_logger": "tensorboard",
12
+ "save_on_interrupt": true,
13
+ "log_model_step": null,
14
+ "save_step": 10000,
15
+ "save_n_checkpoints": 5,
16
+ "save_checkpoints": true,
17
+ "save_all_best": false,
18
+ "save_best_after": 10000,
19
+ "target_loss": "loss_0",
20
+ "print_eval": false,
21
+ "test_delay_epochs": 5,
22
+ "run_eval": true,
23
+ "run_eval_steps": null,
24
+ "distributed_backend": "nccl",
25
+ "distributed_url": "tcp://localhost:54321",
26
+ "mixed_precision": false,
27
+ "precision": "fp16",
28
+ "epochs": 1000,
29
+ "batch_size": 32,
30
+ "eval_batch_size": 16,
31
+ "grad_clip": null,
32
+ "scheduler_after_epoch": true,
33
+ "lr": 0.0001,
34
+ "optimizer": "AdamW",
35
+ "optimizer_params": {
36
+ "betas": [
37
+ 0.8,
38
+ 0.99
39
+ ],
40
+ "weight_decay": 0.0
41
+ },
42
+ "lr_scheduler": null,
43
+ "lr_scheduler_params": {},
44
+ "use_grad_scaler": false,
45
+ "allow_tf32": false,
46
+ "cudnn_enable": true,
47
+ "cudnn_deterministic": false,
48
+ "cudnn_benchmark": false,
49
+ "training_seed": 54321,
50
+ "model": "hifigan",
51
+ "num_loader_workers": 4,
52
+ "num_eval_loader_workers": 4,
53
+ "use_noise_augment": true,
54
+ "audio": {
55
+ "fft_size": 1024,
56
+ "win_length": 1024,
57
+ "hop_length": 256,
58
+ "frame_shift_ms": null,
59
+ "frame_length_ms": null,
60
+ "stft_pad_mode": "reflect",
61
+ "sample_rate": 22050,
62
+ "resample": false,
63
+ "preemphasis": 0.0,
64
+ "ref_level_db": 20,
65
+ "do_sound_norm": false,
66
+ "log_func": "np.log10",
67
+ "do_trim_silence": true,
68
+ "trim_db": 60,
69
+ "do_rms_norm": false,
70
+ "db_level": null,
71
+ "power": 1.5,
72
+ "griffin_lim_iters": 60,
73
+ "num_mels": 80,
74
+ "mel_fmin": 50.0,
75
+ "mel_fmax": null,
76
+ "spec_gain": 1,
77
+ "do_amp_to_db_linear": true,
78
+ "do_amp_to_db_mel": true,
79
+ "pitch_fmax": 640.0,
80
+ "pitch_fmin": 1.0,
81
+ "signal_norm": false,
82
+ "min_level_db": -100,
83
+ "symmetric_norm": true,
84
+ "max_norm": 4.0,
85
+ "clip_norm": true,
86
+ "stats_path": "/Users/thorsten/Library/Application Support/tts/vocoder_models--de--thorsten--hifigan_v1/scale_stats.npy"
87
+ },
88
+ "eval_split_size": 10,
89
+ "data_path": "/home/ubuntu/Thorsten-Voice-Neutral-Dec2021-22kHz",
90
+ "feature_path": "/home/ubuntu/features_330k",
91
+ "seq_len": 8192,
92
+ "pad_short": 2000,
93
+ "conv_pad": 0,
94
+ "use_cache": false,
95
+ "wd": 1e-06,
96
+ "use_stft_loss": false,
97
+ "use_subband_stft_loss": false,
98
+ "use_mse_gan_loss": true,
99
+ "use_hinge_gan_loss": false,
100
+ "use_feat_match_loss": true,
101
+ "use_l1_spec_loss": true,
102
+ "stft_loss_weight": 0.0,
103
+ "subband_stft_loss_weight": 0.0,
104
+ "mse_G_loss_weight": 1.0,
105
+ "hinge_G_loss_weight": 0.0,
106
+ "feat_match_loss_weight": 108.0,
107
+ "l1_spec_loss_weight": 45.0,
108
+ "stft_loss_params": {
109
+ "n_ffts": [
110
+ 1024,
111
+ 2048,
112
+ 512
113
+ ],
114
+ "hop_lengths": [
115
+ 120,
116
+ 240,
117
+ 50
118
+ ],
119
+ "win_lengths": [
120
+ 600,
121
+ 1200,
122
+ 240
123
+ ]
124
+ },
125
+ "l1_spec_loss_params": {
126
+ "use_mel": true,
127
+ "sample_rate": 22050,
128
+ "n_fft": 1024,
129
+ "hop_length": 256,
130
+ "win_length": 1024,
131
+ "n_mels": 80,
132
+ "mel_fmin": 50.0,
133
+ "mel_fmax": null
134
+ },
135
+ "lr_gen": 0.0001,
136
+ "lr_disc": 0.0001,
137
+ "lr_scheduler_gen": "ExponentialLR",
138
+ "lr_scheduler_gen_params": {
139
+ "gamma": 0.999,
140
+ "last_epoch": -1
141
+ },
142
+ "lr_scheduler_disc": "ExponentialLR",
143
+ "lr_scheduler_disc_params": {
144
+ "gamma": 0.999,
145
+ "last_epoch": -1
146
+ },
147
+ "use_pqmf": false,
148
+ "diff_samples_for_G_and_D": false,
149
+ "discriminator_model": "hifigan_discriminator",
150
+ "generator_model": "hifigan_generator",
151
+ "generator_model_params": {
152
+ "upsample_factors": [
153
+ 8,
154
+ 8,
155
+ 2,
156
+ 2
157
+ ],
158
+ "upsample_kernel_sizes": [
159
+ 16,
160
+ 16,
161
+ 4,
162
+ 4
163
+ ],
164
+ "upsample_initial_channel": 512,
165
+ "resblock_kernel_sizes": [
166
+ 3,
167
+ 7,
168
+ 11
169
+ ],
170
+ "resblock_dilation_sizes": [
171
+ [
172
+ 1,
173
+ 3,
174
+ 5
175
+ ],
176
+ [
177
+ 1,
178
+ 3,
179
+ 5
180
+ ],
181
+ [
182
+ 1,
183
+ 3,
184
+ 5
185
+ ]
186
+ ],
187
+ "resblock_type": "1"
188
+ }
189
+ }
vocoder-model_file.pth.tar ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c503e5553714b4fd176c3acaed1b8956815078e9521a4a782dd340952f32ce5f
3
+ size 1016383933