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

View File

@ -1,12 +1,12 @@
import os
from TTS.tts.configs import AlignTTSConfig
from TTS.tts.configs import BaseDatasetConfig
from TTS.trainer import init_training, Trainer, TrainingArgs
from TTS.trainer import Trainer, TrainingArgs, init_training
from TTS.tts.configs import AlignTTSConfig, BaseDatasetConfig
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(
batch_size=32,
eval_batch_size=16,
@ -23,7 +23,7 @@ config = AlignTTSConfig(
print_eval=True,
mixed_precision=False,
output_path=output_path,
datasets=[dataset_config]
datasets=[dataset_config],
)
args, config, output_path, _, c_logger, tb_logger = init_training(TrainingArgs(), config)
trainer = Trainer(args, config, output_path, c_logger, tb_logger)

View File

@ -1,12 +1,12 @@
import os
from TTS.tts.configs import GlowTTSConfig
from TTS.tts.configs import BaseDatasetConfig
from TTS.trainer import init_training, Trainer, TrainingArgs
from TTS.trainer import Trainer, TrainingArgs, init_training
from TTS.tts.configs import BaseDatasetConfig, GlowTTSConfig
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(
batch_size=32,
eval_batch_size=16,
@ -23,7 +23,7 @@ config = GlowTTSConfig(
print_eval=True,
mixed_precision=False,
output_path=output_path,
datasets=[dataset_config]
datasets=[dataset_config],
)
args, config, output_path, _, c_logger, tb_logger = init_training(TrainingArgs(), config)
trainer = Trainer(args, config, output_path, c_logger, tb_logger)

View File

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

View File

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

View File

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

View File

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