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

Update code style in recipes

This commit is contained in:
Eren Gölge 2021-08-07 21:24:50 +00:00
parent 0b8016363a
commit b7f387b3dd
6 changed files with 15 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
recipes/ljspeech/align_tts/train_aligntts.py
View File

@ -1,12 +1,12 @@
import os import os
from TTS.tts.configs import AlignTTSConfig from TTS.trainer import Trainer, TrainingArgs, init_training
from TTS.tts.configs import BaseDatasetConfig from TTS.tts.configs import AlignTTSConfig, BaseDatasetConfig
from TTS.trainer import init_training, Trainer, TrainingArgs
output_path = os.path.dirname(os.path.abspath(__file__)) output_path = os.path.dirname(os.path.abspath(__file__))
dataset_config = BaseDatasetConfig(name="ljspeech", meta_file_train="metadata.csv", path=os.path.join(output_path, "../LJSpeech-1.1/")) dataset_config = BaseDatasetConfig(
name="ljspeech", meta_file_train="metadata.csv", path=os.path.join(output_path, "../LJSpeech-1.1/")
)
config = AlignTTSConfig( config = AlignTTSConfig(
batch_size=32, batch_size=32,
eval_batch_size=16, eval_batch_size=16,
@ -23,7 +23,7 @@ config = AlignTTSConfig(
print_eval=True, print_eval=True,
mixed_precision=False, mixed_precision=False,
output_path=output_path, output_path=output_path,
datasets=[dataset_config] datasets=[dataset_config],
) )
args, config, output_path, _, c_logger, tb_logger = init_training(TrainingArgs(), config) args, config, output_path, _, c_logger, tb_logger = init_training(TrainingArgs(), config)
trainer = Trainer(args, config, output_path, c_logger, tb_logger) trainer = Trainer(args, config, output_path, c_logger, tb_logger)

12
recipes/ljspeech/glow_tts/train_glowtts.py
View File

@ -1,12 +1,12 @@
import os import os
from TTS.tts.configs import GlowTTSConfig from TTS.trainer import Trainer, TrainingArgs, init_training
from TTS.tts.configs import BaseDatasetConfig from TTS.tts.configs import BaseDatasetConfig, GlowTTSConfig
from TTS.trainer import init_training, Trainer, TrainingArgs
output_path = os.path.dirname(os.path.abspath(__file__)) output_path = os.path.dirname(os.path.abspath(__file__))
dataset_config = BaseDatasetConfig(name="ljspeech", meta_file_train="metadata.csv", path=os.path.join(output_path, "../LJSpeech-1.1/")) dataset_config = BaseDatasetConfig(
name="ljspeech", meta_file_train="metadata.csv", path=os.path.join(output_path, "../LJSpeech-1.1/")
)
config = GlowTTSConfig( config = GlowTTSConfig(
batch_size=32, batch_size=32,
eval_batch_size=16, eval_batch_size=16,
@ -23,7 +23,7 @@ config = GlowTTSConfig(
print_eval=True, print_eval=True,
mixed_precision=False, mixed_precision=False,
output_path=output_path, output_path=output_path,
datasets=[dataset_config] datasets=[dataset_config],
) )
args, config, output_path, _, c_logger, tb_logger = init_training(TrainingArgs(), config) args, config, output_path, _, c_logger, tb_logger = init_training(TrainingArgs(), config)
trainer = Trainer(args, config, output_path, c_logger, tb_logger) trainer = Trainer(args, config, output_path, c_logger, tb_logger)

3
recipes/ljspeech/multiband_melgan/train_multiband_melgan.py
View File

@ -1,8 +1,7 @@
import os import os
from TTS.trainer import Trainer, TrainingArgs, init_training
from TTS.vocoder.configs import MultibandMelganConfig from TTS.vocoder.configs import MultibandMelganConfig
from TTS.trainer import init_training, Trainer, TrainingArgs
output_path = os.path.dirname(os.path.abspath(__file__)) output_path = os.path.dirname(os.path.abspath(__file__))
config = MultibandMelganConfig( config = MultibandMelganConfig(

1
recipes/ljspeech/univnet/train.py
View File

@ -1,6 +1,5 @@
import os import os
from TTS.config.shared_configs import BaseAudioConfig
from TTS.trainer import Trainer, TrainingArgs, init_training from TTS.trainer import Trainer, TrainingArgs, init_training
from TTS.vocoder.configs import UnivnetConfig from TTS.vocoder.configs import UnivnetConfig

4
recipes/ljspeech/wavegrad/train_wavegrad.py
View File

@ -1,10 +1,8 @@
import os import os
from TTS.trainer import Trainer, init_training from TTS.trainer import Trainer, TrainingArgs, init_training
from TTS.trainer import TrainingArgs
from TTS.vocoder.configs import WavegradConfig from TTS.vocoder.configs import WavegradConfig
output_path = os.path.dirname(os.path.abspath(__file__)) output_path = os.path.dirname(os.path.abspath(__file__))
config = WavegradConfig( config = WavegradConfig(
batch_size=32, batch_size=32,

3
recipes/ljspeech/wavernn/train_wavernn.py
View File

@ -1,9 +1,8 @@
import os import os
from TTS.trainer import Trainer, init_training, TrainingArgs from TTS.trainer import Trainer, TrainingArgs, init_training
from TTS.vocoder.configs import WavernnConfig from TTS.vocoder.configs import WavernnConfig
output_path = os.path.dirname(os.path.abspath(__file__)) output_path = os.path.dirname(os.path.abspath(__file__))
config = WavernnConfig( config = WavernnConfig(
batch_size=64, batch_size=64,