Commit Graph

4201 Commits

Author SHA1 Message Date
Eren Gölge
5911eec3b1 Small trainer refactoring
1. Use a single Gradscaler for all the optimizers
2. Save terminal logs to a file. In DDP mode, each worker creates `trainer_N_log.txt`.
3. Fixes to allow only the main worker (rank==0) writing to Tensorboard
4. Pass parameters owned by the target optimizer to the grad_clip_norm
2021-08-26 17:08:58 +00:00
fijipants
e9e01b09b0 Fix bug with log_func 2021-08-18 19:59:51 -04:00
fijipants
8f57f8adfd Update synthesizer.py 2021-08-18 19:56:52 -04:00
Eren Gölge
3ab8cef99e Fix VITS model SPD 2021-08-18 14:55:46 +00:00
Eren Gölge
625ab614f0 Fix import in CheckSpectrogram.ipynb 2021-08-17 12:12:36 +00:00
Eren Gölge
3fd78241f8 Fix README link 2021-08-17 12:12:36 +00:00
Eren Gölge
c5d1dd9d1b Fix restoring best_loss
Keep the default value if model checkpoint has no `model_loss`
2021-08-17 12:12:36 +00:00
Eren Gölge
c8bbcdfd07 Fix test_run for DDP 2021-08-13 19:39:02 +00:00
Eren Gölge
7c0d564965 Syncronize DDP processes 2021-08-13 10:40:50 +00:00
Eren Gölge
ecf5f17dca Fix distribute.py and ddp training 2021-08-12 22:22:32 +00:00
Eren Gölge
c308226707 Update default ja vocoder v0.2.0 2021-08-11 08:29:27 +00:00
Eren Gölge
7777d20047 Merge branch 'dev' into main 2021-08-11 08:28:55 +00:00
Eren Gölge
b02c4fe347 Bump up to v0.2.0 2021-08-11 08:15:39 +00:00
Eren Gölge
01a2b0b5c0 Merge pull request #717 from coqui-ai/dev
v0.2.0
2021-08-11 10:08:04 +02:00
Eren Gölge
863525deb1 Add VITS model recipe for LjSpeech 2021-08-10 16:27:29 +00:00
Eren Gölge
537bc8487a Print model count when listing modelsk 2021-08-10 16:25:11 +00:00
Eren Gölge
09ed8426e8 Add the models released with v0.2.0 2021-08-10 15:46:31 +00:00
Eren Gölge
39004484b9 Fix 🐛
Fix synthesizer multi-speaker init
Fix #712
2021-08-10 12:56:32 +00:00
Eren Gölge
c8b9ca3d71 Fix Tacotron num_char init 2021-08-10 08:56:34 +00:00
Eren Gölge
7eb94f760b Remove Ruslan model 2021-08-09 21:48:36 +00:00
Eren Gölge
6af03ac476 Fix num_char init in Tacotron models 2021-08-09 21:46:15 +00:00
Ayush Chaurasia
e685ddfca7 Update trainer.py 2021-08-09 18:37:46 +00:00
Ayush Chaurasia
28870f8df4 update docstring 2021-08-09 18:35:35 +00:00
Ayush Chaurasia
8a246cbb66 Update trainer.py 2021-08-09 18:35:08 +00:00
Ayush Chaurasia
f3e9d61330 Refactor logging initialization 2021-08-09 18:35:08 +00:00
Ayush Chaurasia
79b74a989d Update: add_text 2021-08-09 18:34:38 +00:00
Ayush Chaurasia
9fcf48b760 Delete logger_base.py 2021-08-09 18:34:00 +00:00
Ayush Chaurasia
290972fd35 reformat 2021-08-09 18:34:00 +00:00
Ayush Chaurasia
936a47504d Update Logger API, recipes 2021-08-09 18:34:00 +00:00
Ayush Chaurasia
f63cf46c55 Unified logger API 2021-08-09 18:34:00 +00:00
Ayush Chaurasia
f4434da5a3 Update disabled structure 2021-08-09 18:31:16 +00:00
Ayush Chaurasia
f606741dc4 Add artifacts logging , wandb args 2021-08-09 18:31:16 +00:00
Ayush Chaurasia
f5e50ad502 WandbLogger 2021-08-09 18:27:06 +00:00
Eren Gölge
06018251e6 Add VITS and GlowTTS class docs 🗒️ 2021-08-09 18:02:36 +00:00
Eren Gölge
7a374043cf Add linkify to docs extensions 2021-08-09 18:02:36 +00:00
Eren Gölge
6a7275881d Add VitsConfig docstring 2021-08-09 18:02:36 +00:00
Eren Gölge
9f40172023 Add VITS to README.md 2021-08-09 18:02:36 +00:00
Eren Gölge
f7a72552f1 Make duration predictor dropout configurable 2021-08-09 18:02:36 +00:00
Eren Gölge
922142428c Fix link in the README 2021-08-09 18:02:36 +00:00
Eren Gölge
c312acac7d Implement VITS model 🚀
VITS model implementation built on Glow TTS and HiFiGAN
layers.
2021-08-09 18:02:36 +00:00
Eren Gölge
060e746e21 Add do_amp_to_db option 2021-08-09 18:02:36 +00:00
Eren Gölge
e94c1f894d Simplify console_logger 2021-08-09 18:02:36 +00:00
Eren Gölge
dd55960732 Update synthesizer.py
Fixes and changes for multi-speaker model init and custom symbols  made
by mode.make_symbols()
2021-08-09 18:02:36 +00:00
Eren Gölge
232a5abb6a Update tts.setup_model
Run `model.make_symbols()` if availabe to set the symbol list
2021-08-09 18:02:36 +00:00
Eren Gölge
f5a6aa974f Modify symbols.py not to add _arpanet 2021-08-09 18:02:36 +00:00
Eren Gölge
d4deb2716f Modify get_optimizer to accept a model argument 2021-08-09 18:02:36 +00:00
Eren Gölge
003e5579e8 Enable custom_symbols in text processing
Models can define their own custom symbols lists with custom
`make_symbols()`
2021-08-09 18:02:36 +00:00
Eren Gölge
bd4e29b4dd Add compute_linear_spec=False to BaseTTSConfig 2021-08-09 18:02:36 +00:00
Eren Gölge
960a35a121 Add scheduler_after_epoch to BaseTrainingConfig 2021-08-09 18:02:36 +00:00
Eren Gölge
e4648ffef1 Fix multi-speaker init of Tacotron models & tests 2021-08-09 18:02:36 +00:00