Commit Graph

4034 Commits

Author SHA1 Message Date
Eren Gölge
7ec5c31898 Merge branch 'univnet' into trainer-api 2021-06-29 10:27:12 +02:00
Eren Gölge
828e1265a6 Update torch hub 2021-06-28 17:03:47 +02:00
Eren Gölge
51398cd15b Add docstrings and typing for audio.py 2021-06-28 17:03:47 +02:00
Eren Gölge
ae6405bb76 Docstrings for Trainer 2021-06-28 17:03:47 +02:00
Eren Gölge
65958eaa41 Add preliminary sphinx documentation 2021-06-28 17:03:47 +02:00
Eren Gölge
6b265ae8e3 Docstring update 2021-06-28 17:03:47 +02:00
Eren Gölge
ab563ce7cd Start training by config.json using register_config 2021-06-28 17:03:47 +02:00
Eren Gölge
b3c073c99b Allow runing full path scripts with distribute.py 2021-06-28 17:03:47 +02:00
Eren Gölge
d42d1c02ea Use torch.linalg.qr for pytorch > v1.9.0 2021-06-28 17:03:47 +02:00
Eren Gölge
0a1962b583 Update umap and numba vers. to 0.5.1 and 0.53 2021-06-28 17:03:47 +02:00
Eren Gölge
fbba37e01e Fix loading the amp scaler from a checkpoint 🛠️ 2021-06-28 17:03:47 +02:00
Eren Gölge
a7617d8ab6 Add 🐍 python 3.9 to CI 2021-06-28 17:03:47 +02:00
Eren Gölge
9790eddada Fix wrong argument name 🛠️ 2021-06-28 17:03:47 +02:00
Eren Gölge
7c614525d9 Add AlignTTS recipe 👩‍🍳 2021-06-28 17:03:47 +02:00
Eren Gölge
932ab107ae Docstring edit in TTSDataset.py ✍️ 2021-06-28 17:03:47 +02:00
Eren Gölge
cfa5041db7 Fix eval_log for gan.py 🛠️ 2021-06-28 17:03:47 +02:00
Eren Gölge
d700845b10 Move TorchSTFT to utils.audio 2021-06-28 17:03:47 +02:00
Eren Gölge
5b89cb4fec Fixup trainer.py 🛠️ 2021-06-28 17:03:47 +02:00
Eren Gölge
418c7d98d5 Create LJSpeech recipes for all the models 2021-06-28 17:03:47 +02:00
Eren Gölge
0636c91919 Update gitignore 2021-06-28 17:03:47 +02:00
Eren Gölge
a196007062 Update Pylint configuration 2021-06-28 17:03:47 +02:00
Eren Gölge
8c74f054f0 Enable support for 🐍 python 3.10
Bump up versions numpy 1.19.5 and TF 2.5.0
2021-06-28 17:03:47 +02:00
Eren Gölge
9455a2b01e Apply small fixes for API compatibility 2021-06-28 17:03:47 +02:00
Eren Gölge
a5d5bc9063 Print max_decoder_steps when model reaches the limit 2021-06-28 17:03:47 +02:00
Eren Gölge
e30f245e06 Update synthesizer for speaker and model init 2021-06-28 17:03:47 +02:00
Eren Gölge
15fa31b595 fixup configs 2021-06-28 17:03:47 +02:00
Eren Gölge
f23b228e24 Update speaker_manager 2021-06-28 17:03:47 +02:00
Eren Gölge
e53616078a Fixup utils for the trainer 2021-06-28 17:03:47 +02:00
Eren Gölge
106b63d8a9 Update vocoder utils 2021-06-28 17:03:47 +02:00
Eren Gölge
45947acb60 Update TTS.bin scripts for the new API 2021-06-28 17:03:47 +02:00
Eren Gölge
d7225eedb0 Update vocoder datasets and setup_dataset 2021-06-28 17:03:20 +02:00
Eren Gölge
d18198dff8 Implement setup_model for vocoder models 2021-06-28 17:03:20 +02:00
Eren Gölge
e949e7ad58 Update vocoder models 2021-06-28 17:03:19 +02:00
Eren Gölge
51005cdab4 Update tts.models.setup_model 2021-06-28 17:03:19 +02:00
Eren Gölge
7b8c15ac49 Create base 🐸TTS model abstraction for tts models 2021-06-28 17:03:19 +02:00
Eren Gölge
a358f74a52 Update vocoder model configs 2021-06-28 17:03:19 +02:00
Eren Gölge
786170fe7d Update tts model configs 2021-06-28 17:03:19 +02:00
Eren Gölge
626c9d41e6 Update tests for the new trainer API 2021-06-28 17:03:19 +02:00
Eren Gölge
fcfd95669a Update model test configs 2021-06-28 17:03:19 +02:00
Eren Gölge
98298ee671 Implement unified IO utils 2021-06-28 17:03:19 +02:00
Eren Gölge
c7aad884cd Implement unified trainer 2021-06-28 17:03:19 +02:00
Eren Gölge
6d7b5fbcde tts model abstraction with TTSModel 2021-06-28 17:03:19 +02:00
Eren Gölge
d4dbd89752 fix calculation of loader_start_time 2021-06-28 17:03:19 +02:00
Eren Gölge
c754a0e17d TrainerAbstract and related updates for TrainerTTS 2021-06-28 17:03:19 +02:00
Eren Gölge
00c82c516d rename to 2021-06-28 17:03:19 +02:00
Eren Gölge
166f0aeb9a merge if branches with the same implementation 2021-06-28 17:03:19 +02:00
Eren Gölge
03494ad642 adjust distribute.py for the train_tts.py 2021-06-28 17:03:19 +02:00
Eren Gölge
fdfb18d230 downsize melgan test model size 2021-06-28 17:03:19 +02:00
Eren Gölge
25238e0658 fix glow-tts inference() 2021-06-28 17:03:19 +02:00
Eren Gölge
82582993cc use one testing sentence in tts tests 2021-06-28 17:03:19 +02:00