erogol commited on
Commit
c357750
1 Parent(s): d3b3ea6

Upload 2 files

Browse files

Add model files

Files changed (2) hide show
  1. config.json +185 -0
  2. model.pth +3 -0
config.json ADDED
@@ -0,0 +1,185 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "output_path": "/storage/output-glowtts/",
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.0,
77
+ "mel_fmax": 8000.0,
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
+ },
89
+ "use_phonemes": false,
90
+ "phonemizer": null,
91
+ "phoneme_language": null,
92
+ "compute_input_seq_cache": false,
93
+ "text_cleaner": "no_cleaners",
94
+ "enable_eos_bos_chars": true,
95
+ "test_sentences_file": "",
96
+ "phoneme_cache_path": null,
97
+ "characters": {
98
+ "characters_class": "TTS.tts.utils.text.characters.Graphemes",
99
+ "vocab_dict": null,
100
+ "pad": "_",
101
+ "eos": "~",
102
+ "bos": "^",
103
+ "blank": "@",
104
+ "characters": "Iabdfgijklmnprstuvxz\u0254\u025b\u0263\u0268\u026b\u0271\u0282\u0290\u02b2\u02c8\u02d0\u032f\u0361\u03b2",
105
+ "punctuations": "!,.?: -\u2012\u2013\u2014\u2026",
106
+ "phonemes": null,
107
+ "is_unique": true,
108
+ "is_sorted": true
109
+ },
110
+ "add_blank": true,
111
+ "batch_group_size": 0,
112
+ "loss_masking": null,
113
+ "min_audio_len": 1,
114
+ "max_audio_len": Infinity,
115
+ "min_text_len": 1,
116
+ "max_text_len": Infinity,
117
+ "compute_f0": false,
118
+ "compute_energy": false,
119
+ "compute_linear_spec": false,
120
+ "precompute_num_workers": 0,
121
+ "start_by_longest": false,
122
+ "shuffle": false,
123
+ "drop_last": false,
124
+ "datasets": [
125
+ {
126
+ "formatter": "bel_tts_formatter",
127
+ "dataset_name": "",
128
+ "path": "/storage/filtered_dataset/",
129
+ "meta_file_train": "ipa_final_dataset.csv",
130
+ "ignored_speakers": null,
131
+ "language": "",
132
+ "phonemizer": "",
133
+ "meta_file_val": "",
134
+ "meta_file_attn_mask": ""
135
+ }
136
+ ],
137
+ "test_sentences": [],
138
+ "eval_split_max_size": null,
139
+ "eval_split_size": 0.01,
140
+ "use_speaker_weighted_sampler": false,
141
+ "speaker_weighted_sampler_alpha": 1.0,
142
+ "use_language_weighted_sampler": false,
143
+ "language_weighted_sampler_alpha": 1.0,
144
+ "use_length_weighted_sampler": false,
145
+ "length_weighted_sampler_alpha": 1.0,
146
+ "num_chars": 49,
147
+ "encoder_type": "rel_pos_transformer",
148
+ "encoder_params": {
149
+ "kernel_size": 3,
150
+ "dropout_p": 0.1,
151
+ "num_layers": 6,
152
+ "num_heads": 2,
153
+ "hidden_channels_ffn": 768,
154
+ "input_length": null
155
+ },
156
+ "use_encoder_prenet": true,
157
+ "hidden_channels_enc": 192,
158
+ "hidden_channels_dec": 192,
159
+ "hidden_channels_dp": 256,
160
+ "dropout_p_dp": 0.1,
161
+ "dropout_p_dec": 0.05,
162
+ "mean_only": true,
163
+ "out_channels": 80,
164
+ "num_flow_blocks_dec": 12,
165
+ "inference_noise_scale": 0.0,
166
+ "kernel_size_dec": 5,
167
+ "dilation_rate": 1,
168
+ "num_block_layers": 4,
169
+ "num_speakers": 0,
170
+ "c_in_channels": 0,
171
+ "num_splits": 4,
172
+ "num_squeeze": 2,
173
+ "sigmoid_scale": false,
174
+ "d_vector_dim": 0,
175
+ "data_dep_init_steps": 10,
176
+ "style_wav_for_test": null,
177
+ "length_scale": 1.0,
178
+ "use_speaker_embedding": false,
179
+ "speakers_file": null,
180
+ "use_d_vector_file": false,
181
+ "d_vector_file": false,
182
+ "min_seq_len": 3,
183
+ "max_seq_len": 500,
184
+ "r": 1
185
+ }
model.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1620ac8db50be1219fe1208b2edd1b5db4807554ce1987da5dc6330b2afdf213
3
+ size 343626408