slapula commited on
Commit
2ba9ce0
1 Parent(s): 1bd2695

Upload 5 files

Browse files
best_model.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:76d1f09b00b598812376e7de046493e8983b4ca0e78581d98c41e54768eac030
3
+ size 343659225
config.json ADDED
@@ -0,0 +1,192 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "output_path": "/output",
3
+ "logger_uri": null,
4
+ "run_name": "run",
5
+ "project_name": null,
6
+ "run_description": "\ud83d\udc38Coqui trainer run.",
7
+ "print_step": 50,
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": 2,
16
+ "save_checkpoints": true,
17
+ "save_all_best": false,
18
+ "save_best_after": 5000,
19
+ "target_loss": null,
20
+ "print_eval": true,
21
+ "test_delay_epochs": -1,
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": 96,
30
+ "eval_batch_size": 32,
31
+ "grad_clip": 5.0,
32
+ "scheduler_after_epoch": true,
33
+ "lr": 0.001,
34
+ "optimizer": "RAdam",
35
+ "optimizer_params": {
36
+ "betas": [
37
+ 0.9,
38
+ 0.998
39
+ ],
40
+ "weight_decay": 1e-06
41
+ },
42
+ "lr_scheduler": "NoamLR",
43
+ "lr_scheduler_params": {
44
+ "warmup_steps": 4000
45
+ },
46
+ "use_grad_scaler": false,
47
+ "allow_tf32": false,
48
+ "cudnn_enable": true,
49
+ "cudnn_deterministic": false,
50
+ "cudnn_benchmark": false,
51
+ "training_seed": 54321,
52
+ "model": "glow_tts",
53
+ "num_loader_workers": 8,
54
+ "num_eval_loader_workers": 8,
55
+ "use_noise_augment": true,
56
+ "audio": {
57
+ "fft_size": 1024,
58
+ "win_length": 1024,
59
+ "hop_length": 256,
60
+ "frame_shift_ms": null,
61
+ "frame_length_ms": null,
62
+ "stft_pad_mode": "reflect",
63
+ "sample_rate": 22050,
64
+ "resample": false,
65
+ "preemphasis": 0.0,
66
+ "ref_level_db": 20,
67
+ "do_sound_norm": false,
68
+ "log_func": "np.log10",
69
+ "do_trim_silence": true,
70
+ "trim_db": 45,
71
+ "do_rms_norm": false,
72
+ "db_level": null,
73
+ "power": 1.5,
74
+ "griffin_lim_iters": 60,
75
+ "num_mels": 80,
76
+ "mel_fmin": 50,
77
+ "mel_fmax": 8000,
78
+ "spec_gain": 20,
79
+ "do_amp_to_db_linear": true,
80
+ "do_amp_to_db_mel": true,
81
+ "pitch_fmax": 640.0,
82
+ "pitch_fmin": 1.0,
83
+ "signal_norm": true,
84
+ "min_level_db": -100,
85
+ "symmetric_norm": true,
86
+ "max_norm": 4.0,
87
+ "clip_norm": true,
88
+ "stats_path": "scale_stats_female_2.npy"
89
+ },
90
+ "use_phonemes": false,
91
+ "phonemizer": null,
92
+ "phoneme_language": "be",
93
+ "compute_input_seq_cache": false,
94
+ "text_cleaner": "no_cleaners",
95
+ "enable_eos_bos_chars": true,
96
+ "test_sentences_file": "",
97
+ "phoneme_cache_path": "/output/phoneme_cache",
98
+ "characters": {
99
+ "characters_class": "TTS.tts.utils.text.characters.Graphemes",
100
+ "vocab_dict": null,
101
+ "pad": "_",
102
+ "eos": "~",
103
+ "bos": "^",
104
+ "blank": "@",
105
+ "characters": "Iabdfgijklmnprstuvxz\u0254\u025b\u0263\u0268\u026b\u0271\u0282\u0290\u02b2\u02c8\u02d0\u032f\u0361\u03b2",
106
+ "punctuations": "!,.?: -\u2012\u2013\u2014\u2026",
107
+ "phonemes": null,
108
+ "is_unique": true,
109
+ "is_sorted": true
110
+ },
111
+ "add_blank": true,
112
+ "batch_group_size": 0,
113
+ "loss_masking": null,
114
+ "min_audio_len": 1,
115
+ "max_audio_len": Infinity,
116
+ "min_text_len": 1,
117
+ "max_text_len": Infinity,
118
+ "compute_f0": false,
119
+ "compute_energy": false,
120
+ "compute_linear_spec": false,
121
+ "precompute_num_workers": 0,
122
+ "start_by_longest": false,
123
+ "shuffle": false,
124
+ "drop_last": false,
125
+ "datasets": [
126
+ {
127
+ "formatter": "bel_tts_formatter",
128
+ "dataset_name": "",
129
+ "path": "/output/dataset/female_2",
130
+ "meta_file_train": "updated_metadata.csv",
131
+ "ignored_speakers": null,
132
+ "language": "",
133
+ "phonemizer": "",
134
+ "meta_file_val": "",
135
+ "meta_file_attn_mask": ""
136
+ }
137
+ ],
138
+ "test_sentences": [
139
+ "\u02c8a \u02c8\u0282t\u0254 ja\u02c8mu ja\u02c8\u0282t\u0361\u0282\u025b zasta\u02c8j\u025bt\u0361s\u02d0a?",
140
+ "\u02c8part\u0268ji pa\u02c8v\u02b2in\u02d0\u0268 \u02c8b\u0268t\u0361s\u02b2, naza\u02c8v\u02b2it\u0361s\u02b2\u025b \u02c8mn\u02b2\u025b \u02c8x\u0254t\u0361s\u02b2 a\u02c8dnu i\u02c8snujut\u0361\u0282uju \u02c8part\u0268ju.",
141
+ "\u02c8k\u02b2im b\u0268\u02c8l\u02b2i ja\u02c8n\u0268, k\u02b2irau\u032fn\u02b2i\u02c8k\u02b2i pau\u032f\u02c8stan\u02b2\u02d0a?",
142
+ "xudar\u02c8l\u02b2av\u0268 \u02c8j\u0254n \u02c8b\u0268u\u032f zau\u032f\u02c8s\u02b2\u0254d\u0268 \u02c8ji \u02c8s\u02b2i\u026bu \u02c8m\u02b2\u025bu\u032f \u02c8u pl\u02b2a\u02c8t\u0361\u0282ax"
143
+ ],
144
+ "eval_split_max_size": null,
145
+ "eval_split_size": 0.01,
146
+ "use_speaker_weighted_sampler": false,
147
+ "speaker_weighted_sampler_alpha": 1.0,
148
+ "use_language_weighted_sampler": false,
149
+ "language_weighted_sampler_alpha": 1.0,
150
+ "use_length_weighted_sampler": false,
151
+ "length_weighted_sampler_alpha": 1.0,
152
+ "num_chars": 49,
153
+ "encoder_type": "rel_pos_transformer",
154
+ "encoder_params": {
155
+ "kernel_size": 3,
156
+ "dropout_p": 0.1,
157
+ "num_layers": 6,
158
+ "num_heads": 2,
159
+ "hidden_channels_ffn": 768,
160
+ "input_length": null
161
+ },
162
+ "use_encoder_prenet": true,
163
+ "hidden_channels_enc": 192,
164
+ "hidden_channels_dec": 192,
165
+ "hidden_channels_dp": 256,
166
+ "dropout_p_dp": 0.1,
167
+ "dropout_p_dec": 0.05,
168
+ "mean_only": true,
169
+ "out_channels": 80,
170
+ "num_flow_blocks_dec": 12,
171
+ "inference_noise_scale": 0.0,
172
+ "kernel_size_dec": 5,
173
+ "dilation_rate": 1,
174
+ "num_block_layers": 4,
175
+ "num_speakers": 0,
176
+ "c_in_channels": 0,
177
+ "num_splits": 4,
178
+ "num_squeeze": 2,
179
+ "sigmoid_scale": false,
180
+ "d_vector_dim": 0,
181
+ "data_dep_init_steps": 10,
182
+ "style_wav_for_test": null,
183
+ "length_scale": 1.0,
184
+ "use_speaker_embedding": false,
185
+ "speakers_file": null,
186
+ "use_d_vector_file": false,
187
+ "d_vector_file": false,
188
+ "min_seq_len": 3,
189
+ "max_seq_len": 500,
190
+ "r": 1,
191
+ "github_branch": "* dev"
192
+ }
scale_stats_female_2.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:05c5f83e0b9529a3496a368add882343ad168f7e9a5e959dceccb54223019ed1
3
+ size 10797
vocoder_best_model.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d8a5f2a3e1bbb76c4dc6e0e74ab275497a2ade18e04feb6840315a8f3a2b36b9
3
+ size 1016515818
vocoder_config.json ADDED
@@ -0,0 +1,191 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "output_path": "/output/vocoder",
3
+ "logger_uri": null,
4
+ "run_name": "run",
5
+ "project_name": null,
6
+ "run_description": "\ud83d\udc38Coqui trainer run.",
7
+ "print_step": 50,
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": 5000,
15
+ "save_n_checkpoints": 2,
16
+ "save_checkpoints": true,
17
+ "save_all_best": false,
18
+ "save_best_after": 5000,
19
+ "target_loss": "loss_0",
20
+ "print_eval": true,
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": 74,
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": 8,
52
+ "num_eval_loader_workers": 8,
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": 45,
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": 8000.0,
76
+ "spec_gain": 20,
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": true,
82
+ "min_level_db": -100,
83
+ "symmetric_norm": true,
84
+ "max_norm": 4.0,
85
+ "clip_norm": true,
86
+ "stats_path": "scale_stats_female_2.npy"
87
+ },
88
+ "eval_split_size": 30,
89
+ "data_path": "/output/dataset/female_2",
90
+ "feature_path": null,
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": 0.0,
133
+ "mel_fmax": null
134
+ },
135
+ "lr_gen": 0.0002,
136
+ "lr_disc": 0.0002,
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
+ "restore_path": "/output/vocoder/run-December-31-2023_07+52PM-55c70637/best_model_5060.pth",
190
+ "github_branch": "* dev"
191
+ }