From 843b3ba9603bd5219b873922381ddc1ce792d969 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Wed, 26 May 2021 16:02:01 +0200 Subject: [PATCH] update `speedy_speecy_config.py` for the trainer --- TTS/tts/configs/speedy_speech_config.py | 17 ++++++++++++++--- 1 file changed, 14 insertions(+), 3 deletions(-) diff --git a/TTS/tts/configs/speedy_speech_config.py b/TTS/tts/configs/speedy_speech_config.py index 1b8f0c82..42258398 100644 --- a/TTS/tts/configs/speedy_speech_config.py +++ b/TTS/tts/configs/speedy_speech_config.py @@ -1,4 +1,5 @@ from dataclasses import dataclass, field +from typing import List from TTS.tts.configs.shared_configs import BaseTTSConfig @@ -99,10 +100,11 @@ class SpeedySpeechConfig(BaseTTSConfig): external_speaker_embedding_file: str = False # optimizer parameters - noam_schedule: bool = False - warmup_steps: int = 4000 + optimizer: str = "RAdam" + optimizer_params: dict = field(default_factory=lambda: {'betas': [0.9, 0.998], 'weight_decay': 1e-6}) + lr_scheduler: str = None + lr_scheduler_params: dict = None lr: float = 1e-4 - wd: float = 1e-6 grad_clip: float = 5.0 # loss params @@ -114,3 +116,12 @@ class SpeedySpeechConfig(BaseTTSConfig): min_seq_len: int = 13 max_seq_len: int = 200 r: int = 1 # DO NOT CHANGE + + # testing + test_sentences: List[str] = field(default_factory=lambda:[ + "It took me quite a long time to develop a voice, and now that I have it I'm not going to be silent.", + "Be a voice, not an echo.", + "I'm sorry Dave. I'm afraid I can't do that.", + "This cake is great. It's so delicious and moist.", + "Prior to November 22, 1963." + ])