Commit Graph

15 Commits

Author SHA1 Message Date
Feiyu Chan 4f288a6d4f
add ge2e and tacotron2_aishell3 example (#107)
* hacky thing, add tone support for acoustic model

* fix experiments for waveflow and wavenet, only write visual log in rank-0

* use emb add in tacotron2

* 1. remove space from numericalized representation;
2. fix decoder paddign mask's unsqueeze dim.

* remove bn in postnet

* refactoring code

* add an option to normalize volume when loading audio.

* add an embedding layer.

* 1. change the default min value of LogMagnitude to 1e-5;
2. remove stop logit prediction from tacotron2 model.

* WIP: baker

* add ge2e

* fix lstm speaker encoder

* fix lstm speaker encoder

* fix speaker encoder and add support for 2 more datasets

* simplify visualization code

* add a simple strategy to support multispeaker for tacotron.

* add vctk example for refactored tacotron

* fix indentation

* fix class name

* fix visualizer

* fix root path

* fix root path

* fix root path

* fix typos

* fix bugs

* fix text log extention name

* add example for baker and aishell3

* update experiment and display

* format code for tacotron_vctk, add plot_waveform to display

* add new trainer

* minor fix

* add global condition support for tacotron2

* add gst layer

* add 2 frontend

* fix fmax for example/waveflow

* update collate function, data loader not does not convert nested list into numpy array.

* WIP: add hifigan

* WIP:update hifigan

* change stft to use conv1d

* add audio datasets

* change batch_text_id, batch_spec, batch_wav to include valid lengths in the returned value

* change wavenet to use on-the-fly prepeocessing

* fix typos

* resolve conflict

* remove imports that are removed

* remove files not included in this release

* remove imports to deleted modules

* move tacotron2_msp

* clean code

* fix argument order

* fix argument name

* clean code for data processing

* WIP: add README

* add more details to thr README, fix some preprocess scripts

* add voice cloning notebook

* add an optional to alter the loss and model structure of tacotron2, add an alternative config

* add plot_multiple_attentions and update visualization code in transformer_tts

* format code

* remove tacotron2_msp

* update tacotron2 from_pretrained, update setup.py

* update tacotron2

* update tacotron_aishell3's README

* add images for exampels/tacotron2_aishell3's README

* update README for examples/ge2e

* add STFT back

* add extra_config keys into the default config of tacotron

* fix typos and docs

* update README and doc

* update docstrings for tacotron

* update doc

* update README

* add links to downlaod pretrained models

* refine READMEs and clean code

* add praatio into requirements for running the experiments

* format code with pre-commit

* simplify text processing code and update notebook
2021-05-13 17:49:50 +08:00
chenfeiyu 6f1c534557 use exist_ok to ensure no conflict 2021-02-23 10:50:57 +08:00
chenfeiyu e69ab88fe6 fix multiprocessing training: other processes have to wait untils the output directory in created. 2021-02-23 10:40:14 +08:00
Hui Zhang c955c4192b fix tensorboard error https://github.com/pytorch/fairseq/issues/2357 2021-02-18 19:53:58 +08:00
chenfeiyu 0af7402daa add rank_zero_only for ExperimentBase.save 2021-02-18 19:33:41 +08:00
chenfeiyu f423323bae fix bugs with multiprocess training. 2021-02-18 19:09:54 +08:00
iclementine e03e96d9e4 format all the code with yapf 2020-12-20 13:15:07 +08:00
iclementine 84ad4c9e65 1. update docstrings for models.wavenet;
2. remove unnecessary code;
3. fix typos
2020-12-18 20:55:27 +08:00
iclementine bbc50faef2 add generated api_doc 2020-12-18 10:54:50 +08:00
Feiyu Chan 4de58f4a99
Merge pull request #48 from iclementine/reborn
add set_device at experiment setup
2020-12-11 19:50:27 +08:00
chenfeiyu a079e767df add set_device at experiment setup 2020-12-11 19:45:49 +08:00
Feiyu Chan 1d2e93c75f
Merge pull request #45 from lfchener/reborn
add TTS model tacotron2
2020-12-11 16:33:22 +08:00
chenfeiyu cecc8735c4 1. fix ExperimentBase, create ouput folder with parent,
2. fix stop condition
2020-12-11 11:58:57 +08:00
lfchener 09f1840082 fix some bugs of tacotron2 2020-12-11 03:56:40 +00:00
chenfeiyu 4893c9c086 add an ExperimentBase class and default config for training 2020-12-08 15:46:41 +08:00