jhlfrfufyfn commited on
Commit
68d7e5a
1 Parent(s): 81af215

Upload 5 files

Browse files
Files changed (5) hide show
  1. config.json +183 -0
  2. model.pth +3 -0
  3. scale_stats.npy +3 -0
  4. vocoder-config.json +187 -0
  5. vocoder.pth +3 -0
config.json ADDED
@@ -0,0 +1,183 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "output_path": "/storage/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
+ "log_model_step": null,
13
+ "save_step": 10000,
14
+ "save_n_checkpoints": 2,
15
+ "save_checkpoints": true,
16
+ "save_all_best": false,
17
+ "save_best_after": 5000,
18
+ "target_loss": null,
19
+ "print_eval": true,
20
+ "test_delay_epochs": -1,
21
+ "run_eval": true,
22
+ "distributed_backend": "nccl",
23
+ "distributed_url": "tcp://localhost:54321",
24
+ "mixed_precision": false,
25
+ "epochs": 1000,
26
+ "batch_size": 64,
27
+ "eval_batch_size": 32,
28
+ "grad_clip": 5.0,
29
+ "scheduler_after_epoch": true,
30
+ "lr": 0.001,
31
+ "optimizer": "RAdam",
32
+ "optimizer_params": {
33
+ "betas": [
34
+ 0.9,
35
+ 0.998
36
+ ],
37
+ "weight_decay": 1e-06
38
+ },
39
+ "lr_scheduler": "ExponentialLR",
40
+ "lr_scheduler_params": {
41
+ "gamma": 0.95,
42
+ "last_epoch": -1
43
+ },
44
+ "use_grad_scaler": false,
45
+ "cudnn_enable": true,
46
+ "cudnn_deterministic": false,
47
+ "cudnn_benchmark": true,
48
+ "training_seed": 54321,
49
+ "model": "glow_tts",
50
+ "num_loader_workers": 8,
51
+ "num_eval_loader_workers": 8,
52
+ "use_noise_augment": true,
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": 16000,
61
+ "resample": false,
62
+ "preemphasis": 0.0,
63
+ "ref_level_db": 20,
64
+ "do_sound_norm": false,
65
+ "log_func": "np.log10",
66
+ "do_trim_silence": true,
67
+ "trim_db": 45,
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": 8000.0,
75
+ "spec_gain": 1,
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": true,
81
+ "min_level_db": -100,
82
+ "symmetric_norm": true,
83
+ "max_norm": 4.0,
84
+ "clip_norm": true,
85
+ "stats_path": "/app/tts-config/scale_stats.npy"
86
+ },
87
+ "use_phonemes": false,
88
+ "phonemizer": null,
89
+ "phoneme_language": null,
90
+ "compute_input_seq_cache": false,
91
+ "text_cleaner": "belarusian_cleaners",
92
+ "enable_eos_bos_chars": true,
93
+ "test_sentences_file": "",
94
+ "phoneme_cache_path": null,
95
+ "characters": {
96
+ "characters_class": "TTS.tts.utils.text.characters.Graphemes",
97
+ "vocab_dict": null,
98
+ "pad": "_",
99
+ "eos": "~",
100
+ "bos": "^",
101
+ "blank": "@",
102
+ "characters": "\u0430\u0431\u0432\u0433\u0434\u0435\u0451\u0436\u0437\u0456\u0439\u043a\u043b\u043c\u043d\u043e\u043f\u0440\u0441\u0442\u0443\u045e\u0444\u0445\u0446\u0447\u0448\u044b\u044c\u044d\u044e\u044f'",
103
+ "punctuations": "!,.? \u2014",
104
+ "phonemes": null,
105
+ "is_unique": true,
106
+ "is_sorted": true
107
+ },
108
+ "add_blank": true,
109
+ "batch_group_size": 0,
110
+ "loss_masking": null,
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": 0,
119
+ "start_by_longest": false,
120
+ "datasets": [
121
+ {
122
+ "name": "commonvoice-be",
123
+ "path": "/storage/be",
124
+ "meta_file_train": "metadata.csv",
125
+ "ignored_speakers": null,
126
+ "language": "",
127
+ "meta_file_val": "",
128
+ "meta_file_attn_mask": ""
129
+ }
130
+ ],
131
+ "test_sentences": [
132
+ "\u0422\u044d\u0441\u0442\u0430\u0432\u044b \u0441\u043a\u0430\u0437.",
133
+ "\u0423 \u0440\u0443\u0434\u043e\u0433\u0430 \u0432\u0435\u0440\u0430\u0431\u2019\u044f \u045e \u0441\u0445\u043e\u0432\u0456\u0448\u0447\u044b \u043f\u0430\u0434 \u0444\u0430\u0442\u044d\u043b\u0435\u043c \u043b\u044f\u0436\u0430\u0446\u044c \u043d\u0435\u0439\u043a\u0456\u044f \u0433\u0430\u044e\u0447\u044b\u044f \u0437\u0451\u043b\u043a\u0456",
134
+ "\u042f \u0436\u043e\u0440\u0441\u0442\u043a\u0430 \u0437\u0430\u0431\u2019\u044e \u043f\u0440\u043e\u0441\u0442\u0430 \u045e \u0441\u044d\u0440\u0446\u0430 \u0433\u044d\u0442\u044b \u0440\u0430\u0441\u043a\u0432\u0435\u0447\u0430\u043d\u044b \u043f\u0440\u043e\u0444\u0456\u043b\u044c, \u0448\u0442\u043e \u0445\u043e\u0434\u0437\u0456\u0446\u044c \u043b\u044f \u043c\u0430\u0451\u0439 \u0445\u0430\u0442\u044b"
135
+ ],
136
+ "eval_split_max_size": null,
137
+ "eval_split_size": 0.01,
138
+ "use_speaker_weighted_sampler": false,
139
+ "speaker_weighted_sampler_alpha": 1.0,
140
+ "use_language_weighted_sampler": false,
141
+ "language_weighted_sampler_alpha": 1.0,
142
+ "num_chars": 43,
143
+ "encoder_type": "rel_pos_transformer",
144
+ "encoder_params": {
145
+ "kernel_size": 3,
146
+ "dropout_p": 0.1,
147
+ "num_layers": 6,
148
+ "num_heads": 2,
149
+ "hidden_channels_ffn": 768,
150
+ "input_length": null
151
+ },
152
+ "use_encoder_prenet": true,
153
+ "hidden_channels_enc": 192,
154
+ "hidden_channels_dec": 192,
155
+ "hidden_channels_dp": 256,
156
+ "dropout_p_dp": 0.1,
157
+ "dropout_p_dec": 0.05,
158
+ "mean_only": true,
159
+ "out_channels": 80,
160
+ "num_flow_blocks_dec": 12,
161
+ "inference_noise_scale": 0.0,
162
+ "kernel_size_dec": 5,
163
+ "dilation_rate": 1,
164
+ "num_block_layers": 4,
165
+ "num_speakers": 0,
166
+ "c_in_channels": 0,
167
+ "num_splits": 4,
168
+ "num_squeeze": 2,
169
+ "sigmoid_scale": false,
170
+ "d_vector_dim": 0,
171
+ "data_dep_init_steps": 10,
172
+ "style_wav_for_test": null,
173
+ "length_scale": 1.0,
174
+ "use_speaker_embedding": false,
175
+ "speakers_file": null,
176
+ "use_d_vector_file": false,
177
+ "d_vector_file": false,
178
+ "min_seq_len": 3,
179
+ "max_seq_len": 500,
180
+ "r": 1,
181
+ "restore_path": "/storage/output/newer-glow/checkpoint_70000.pth",
182
+ "github_branch": "* main"
183
+ }
model.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:57bf5d9621527a536ca1439c579e19c7dd85576bf91e6310fe6c59233bc98dc1
3
+ size 343613991
scale_stats.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:573b2b88b380c58cc019e0c54818bdfc4c88f489f3728161df0eb1e70115ef82
3
+ size 10788
vocoder-config.json ADDED
@@ -0,0 +1,187 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "output_path": "/storage/output",
3
+ "logger_uri": null,
4
+ "run_name": "run",
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": 2000,
13
+ "save_step": 2000,
14
+ "save_n_checkpoints": 2,
15
+ "save_checkpoints": true,
16
+ "save_all_best": false,
17
+ "save_best_after": 10000,
18
+ "target_loss": "loss_1",
19
+ "print_eval": true,
20
+ "test_delay_epochs": 5,
21
+ "run_eval": true,
22
+ "distributed_backend": "nccl",
23
+ "distributed_url": "tcp://localhost:54321",
24
+ "mixed_precision": false,
25
+ "epochs": 1000,
26
+ "batch_size": 32,
27
+ "eval_batch_size": 16,
28
+ "grad_clip": null,
29
+ "scheduler_after_epoch": true,
30
+ "lr": 0.0001,
31
+ "optimizer": "AdamW",
32
+ "optimizer_params": {
33
+ "betas": [
34
+ 0.8,
35
+ 0.99
36
+ ],
37
+ "weight_decay": 0.0
38
+ },
39
+ "lr_scheduler": null,
40
+ "lr_scheduler_params": null,
41
+ "use_grad_scaler": false,
42
+ "cudnn_enable": true,
43
+ "cudnn_deterministic": false,
44
+ "cudnn_benchmark": true,
45
+ "training_seed": 54321,
46
+ "model": "hifigan",
47
+ "num_loader_workers": 8,
48
+ "num_eval_loader_workers": 8,
49
+ "use_noise_augment": true,
50
+ "audio": {
51
+ "fft_size": 1024,
52
+ "win_length": 1024,
53
+ "hop_length": 256,
54
+ "frame_shift_ms": null,
55
+ "frame_length_ms": null,
56
+ "stft_pad_mode": "reflect",
57
+ "sample_rate": 16000,
58
+ "resample": false,
59
+ "preemphasis": 0.0,
60
+ "ref_level_db": 20,
61
+ "do_sound_norm": false,
62
+ "log_func": "np.log10",
63
+ "do_trim_silence": true,
64
+ "trim_db": 45,
65
+ "do_rms_norm": false,
66
+ "db_level": null,
67
+ "power": 1.5,
68
+ "griffin_lim_iters": 60,
69
+ "num_mels": 80,
70
+ "mel_fmin": 50.0,
71
+ "mel_fmax": 8000.0,
72
+ "spec_gain": 1,
73
+ "do_amp_to_db_linear": true,
74
+ "do_amp_to_db_mel": true,
75
+ "pitch_fmax": 640.0,
76
+ "pitch_fmin": 0.0,
77
+ "signal_norm": true,
78
+ "min_level_db": -100,
79
+ "symmetric_norm": true,
80
+ "max_norm": 4.0,
81
+ "clip_norm": true,
82
+ "stats_path": "/app/tts-config/scale_stats.npy"
83
+ },
84
+ "eval_split_size": 15,
85
+ "data_path": "/storage/be",
86
+ "feature_path": null,
87
+ "seq_len": 8192,
88
+ "pad_short": 2000,
89
+ "conv_pad": 0,
90
+ "use_cache": false,
91
+ "wd": 1e-06,
92
+ "use_stft_loss": false,
93
+ "use_subband_stft_loss": false,
94
+ "use_mse_gan_loss": true,
95
+ "use_hinge_gan_loss": false,
96
+ "use_feat_match_loss": true,
97
+ "use_l1_spec_loss": true,
98
+ "stft_loss_weight": 0.0,
99
+ "subband_stft_loss_weight": 0.0,
100
+ "mse_G_loss_weight": 1.0,
101
+ "hinge_G_loss_weight": 0.0,
102
+ "feat_match_loss_weight": 108.0,
103
+ "l1_spec_loss_weight": 45.0,
104
+ "stft_loss_params": {
105
+ "n_ffts": [
106
+ 1024,
107
+ 2048,
108
+ 512
109
+ ],
110
+ "hop_lengths": [
111
+ 120,
112
+ 240,
113
+ 50
114
+ ],
115
+ "win_lengths": [
116
+ 600,
117
+ 1200,
118
+ 240
119
+ ]
120
+ },
121
+ "l1_spec_loss_params": {
122
+ "use_mel": true,
123
+ "sample_rate": 16000,
124
+ "n_fft": 1024,
125
+ "hop_length": 256,
126
+ "win_length": 1024,
127
+ "n_mels": 80,
128
+ "mel_fmin": 50.0,
129
+ "mel_fmax": 8000
130
+ },
131
+ "lr_gen": 1e-05,
132
+ "lr_disc": 1e-05,
133
+ "lr_scheduler_gen": "ExponentialLR",
134
+ "lr_scheduler_gen_params": {
135
+ "gamma": 0.999,
136
+ "last_epoch": -1
137
+ },
138
+ "lr_scheduler_disc": "ExponentialLR",
139
+ "lr_scheduler_disc_params": {
140
+ "gamma": 0.999,
141
+ "last_epoch": -1
142
+ },
143
+ "use_pqmf": false,
144
+ "diff_samples_for_G_and_D": false,
145
+ "discriminator_model": "hifigan_discriminator",
146
+ "generator_model": "hifigan_generator",
147
+ "generator_model_params": {
148
+ "upsample_factors": [
149
+ 8,
150
+ 8,
151
+ 2,
152
+ 2
153
+ ],
154
+ "upsample_kernel_sizes": [
155
+ 16,
156
+ 16,
157
+ 4,
158
+ 4
159
+ ],
160
+ "upsample_initial_channel": 512,
161
+ "resblock_kernel_sizes": [
162
+ 3,
163
+ 7,
164
+ 11
165
+ ],
166
+ "resblock_dilation_sizes": [
167
+ [
168
+ 1,
169
+ 3,
170
+ 5
171
+ ],
172
+ [
173
+ 1,
174
+ 3,
175
+ 5
176
+ ],
177
+ [
178
+ 1,
179
+ 3,
180
+ 5
181
+ ]
182
+ ],
183
+ "resblock_type": "1"
184
+ },
185
+ "restore_path": "/storage/output/finetuning-hifigan/checkpoint_286000.pth",
186
+ "github_branch": "* dev"
187
+ }
vocoder.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4e80d2eef7c5b58a174442071e779fee87b1e8a887ce9f18a1c50919afa136f7
3
+ size 1016384188