LightSpeech
1.0.0
LightSpeech的非官方Pytorch实现:通过神经体系结构搜索进行语音轻巧和快速文本。该回购使用ESPNET的FastSpeech 2实现作为基础。该仓库仅实现LightSpeech模型的最终版本,而不是论文中提到的神经体系结构搜索。
But I am able to compress only 3x (from 27 M to 7.99 M trainable parameters) not 15x。
所有代码以Python 3.6.2编写。
在安装Pytorch之前,请通过运行以下命令来检查您的CUDA版本:
nvcc --version
pip install torch torchvision
在此存储库中,我使用了pytorch 1.6.0用于torch.bucketize功能,这在pytorch的先前版本中不存在。
pip install -r requirements.txt
tensorboard version 1.14.0分别使用受支持的tensorflow (1.14.0) filelists文件夹包含MFA(Motreal Force Aligner)处理的LJSpeech数据集文件,因此您无需将文本与LJSpeech数据集的音频(用于提取持续时间)对齐。对于其他数据集,请在此处遵循指令。对于其他预处理运行以下命令:
python .nvidia_preprocessing.py -d path_of_wavs -c configs/default.yaml
查找F0和能量的最小和最大
python .compute_statistics.py
在hparams.py中更新以下内容,按min和最大的f0和能量更新
p_min = Min F0/pitch
p_max = Max F0
e_min = Min energy
e_max = Max energy
python train_lightspeech.py --outdir etc -c configs/default.yaml -n "name"
WIP
python .inference.py -c .configsdefault.yaml -p .checkpointsfirst_1xyz.pyt --out output --text "ModuleList can be indexed like a regular Python list but modules it contains are properly registered."
python export_torchscript.py -c configs/default.yaml -n fastspeech_scrip --outdir etc