Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix lint

This commit is contained in:
Edresson Casanova 2022-12-08 14:01:04 -03:00
parent e87bbdef5d
commit 5d925eaa4d
1 changed files with 12 additions and 3 deletions
recipes/vctk/yourtts
train_yourtts.py

15
recipes/vctk/yourtts/train_yourtts.py
View File

@ -1,14 +1,23 @@
import torch
import os import os
import torch
from trainer import Trainer, TrainerArgs from trainer import Trainer, TrainerArgs
from TTS.config.shared_configs import BaseDatasetConfig from TTS.config.shared_configs import BaseDatasetConfig
from TTS.tts.datasets import load_tts_samples
from TTS.tts.models.vits import VitsArgs, VitsAudioConfig, Vits
from TTS.tts.configs.vits_config import VitsConfig from TTS.tts.configs.vits_config import VitsConfig
from TTS.tts.datasets import load_tts_samples
from TTS.tts.models.vits import Vits, VitsArgs, VitsAudioConfig
torch.set_num_threads(24) torch.set_num_threads(24)
# pylint: disable=W0105
"""
This recipe replicates the first experiment proposed in the YourTTS paper (https://arxiv.org/abs/2112.02418).
YourTTS model is based on the VITS model however it uses external speaker embeddings extracted from a pre-trained speaker encoder and has small architecture changes.
In addition, YourTTS can be trained in multilingual data, however, this recipe replicates the single language training using the VCTK dataset.
The VitsArgs instance has commented parameters used to enable the multilingual training.
"""
# Name of the run for the Trainer # Name of the run for the Trainer
RUN_NAME = "YourTTS-EN-VCTK" RUN_NAME = "YourTTS-EN-VCTK"