|
|
4 years ago | |
|---|---|---|
| .. | ||
| local | 4 years ago | |
| README.md | 4 years ago | |
| path.sh | 4 years ago | |
| run.sh | 4 years ago | |
README.md
WaveFlow with LJSpeech
Dataset
Download and Extract
Download LJSpeech-1.1 from it's Official Website and extract it to ~/datasets. Then the dataset is in the directory ~/datasets/LJSpeech-1.1.
Get Started
Assume the path to the dataset is ~/datasets/LJSpeech-1.1.
Assume the path to the Tacotron2 generated mels is ../tts0/output/test.
Run the command below to
- source path.
- preprocess the dataset.
- train the model.
- synthesize wavs from mels.
./run.sh
You can choose a range of stages you want to run, or set stage equal to stop-stage to use only one stage, for example, running the following command will only preprocess the dataset.
./run.sh --stage 0 --stop-stage 0
Data Preprocessing
./local/preprocess.sh ${preprocess_path}
Model Training
./local/train.sh calls ${BIN_DIR}/train.py.
CUDA_VISIBLE_DEVICES=${gpus} ./local/train.sh ${preprocess_path} ${train_output_path}
The training script requires 4 command line arguments.
--datais the path of the training dataset.--outputis the path of the output directory.--ngpuis the number of gpus to use, if ngpu == 0, use cpu.
If you want distributed training, set a larger --ngpu (e.g. 4). Note that distributed training with cpu is not supported yet.
Synthesizing
./local/synthesize.sh calls ${BIN_DIR}/synthesize.py, which can synthesize waveform from mels.
CUDA_VISIBLE_DEVICES=${gpus} ./local/synthesize.sh ${input_mel_path} ${train_output_path} ${ckpt_name}
Synthesize waveform.
- We assume the
--inputis a directory containing several mel spectrograms(log magnitude) in.npyformat. - The output would be saved in the
--outputdirectory, containing several.wavfiles, each with the same name as the mel spectrogram does. --checkpoint_pathshould be the path of the parameter file (.pdparams) to load. Note that the extention name.pdparmasis not included here.--ngpuis the number of gpus to use, if ngpu == 0, use cpu.
Pretrained Model
Pretrained Model with residual channel equals 128 can be downloaded here: