Commit Graph

245 Commits

Author SHA1 Message Date
chenfeiyu 8dbcc9bccb add profiling 2021-06-16 09:40:47 +00:00
chenfeiyu 95f64c4f02 WIP: add some trainig info 2021-06-14 17:21:45 +08:00
chenfeiyu b0983e4d76 WIP: pwg training works 2021-06-14 17:05:37 +08:00
chenfeiyu 54c7905f40 WIP: training setup done 2021-06-13 17:24:15 +08:00
chenfeiyu 0067851950 fix imports 2021-06-11 05:26:22 +08:00
chenfeiyu 66062d29e5 WIP: add sample code for parallel wavegan 2021-06-11 02:43:10 +08:00
chenfeiyu 258083aea9 fix file names 2021-06-11 01:01:43 +08:00
chenfeiyu f0a5ac8c5a move docstring to class 2021-06-11 00:59:07 +08:00
chenfeiyu 13ab0bd608 remove task-specific Datasets and add a general purpose DataTable 2021-06-10 22:58:53 +08:00
chenfeiyu 3bf2e71734 checkpoint utility: add support for str in addition to Path object 2021-06-10 20:34:59 +08:00
chenfeiyu 60c16dcfb7 add AudioMelDataset for training vocoders. 2021-06-10 20:16:14 +08:00
chenfeiyu 988d6d3268 add h5py utility to interact with numpy 2021-06-10 18:57:46 +08:00
chenfeiyu 6a8b3f92df add AudioDataset and MelDataset 2021-06-10 16:26:09 +08:00
chenfeiyu 3c964fde54 add parallel wavegan model 2021-06-10 04:08:05 +08:00
chenfeiyu 759999c738 STFT and MelScale: register filters as buffer. 2021-06-10 04:06:06 +08:00
chenfeiyu c306f5c2b3 add k-latest 2021-06-10 03:39:54 +08:00
chenfeiyu dc9040dd4d fix typos 2021-06-10 02:45:45 +08:00
chenfeiyu 0114a808a2 fix del fn 2021-06-10 00:40:54 +08:00
chenfeiyu 13323bdf6a remove unnecessary imports 2021-06-10 00:32:49 +08:00
chenfeiyu 37a66f1506 fix typos 2021-06-10 00:29:25 +08:00
chenfeiyu b571b506c3 add kbest 2021-06-10 00:25:00 +08:00
chenfeiyu 3aa6aed0fb fix a config key error 2021-05-18 18:57:13 +08:00
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
chenfeiyu 30e3b9172f 1. fix imports for renamed functions in position encoding;
2. fix dimension in MLPPrenet in transformer_tts;
3. use dropout also in inference in MLPPrenet in transformer_tts.
2021-02-03 14:24:29 +08:00
iclementine c321fcd098 polish documentation 2021-01-13 14:58:26 +08:00
iclementine 641be1bc92 Merge branch 'develop' of github.com:iclementine/Parakeet into doc 2021-01-13 11:09:05 +08:00
iclementine c2a279c433 add documentation sections 2021-01-13 11:06:15 +08:00
lfchener b0ba6e7bf9 fix an encoding problem in windows 2021-01-08 02:47:43 +00:00
lfchener 80bf04b710 fix the behavior of dropout in eval of tacotron2 2020-12-28 08:28:55 +00:00
chenfeiyu 2421a936ed fix positional encoding naming conflict 2020-12-21 17:41:18 +08:00
iclementine bb64e4659a discard opencc untill we find an easy solution to install it on windows 2020-12-20 13:46:45 +08:00
iclementine e03e96d9e4 format all the code with yapf 2020-12-20 13:15:07 +08:00
iclementine f31643b33c 1. fix typos;
2. add tensorboardX into install requirements.
2020-12-19 20:08:25 +08:00
iclementine 18709adce8 update setup.py and version str 2020-12-19 18:55:42 +08:00
iclementine b6efb43990 update docstring for waveflow 2020-12-19 18:33:07 +08:00
iclementine f2a35a17d4 import normalizer into frontend 2020-12-19 16:20:41 +08:00
Li Fuchen 544594ec54
Merge pull request #63 from iclementine/doc
update docstrings for models.wavenet.
2020-12-18 20:57:28 +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
lfchener 255ddcfe32 modified docstring of tacotron2 2020-12-18 20:28:21 +08:00
lfchener 63285dc80f add docstring for normalizer 2020-12-18 19:36:12 +08:00
lfchener c2bc4b0474 add docstring for phonectic and vocab 2020-12-18 19:31:44 +08:00
lfchener 1af9127ee6 add docstring for LocationSensitiveAttention 2020-12-18 17:31:51 +08:00
Feiyu Chan dd2c5cc6c6
Merge pull request #59 from iclementine/doc
update docstrings
2020-12-18 16:12:56 +08:00
iclementine 310366bb54 1. fix format errors and typos 2020-12-18 16:09:38 +08:00
lfchener 3baffa5f4c update link in docstring 2020-12-18 15:53:24 +08:00