mirror of https://github.com/coqui-ai/TTS.git
Update tests
This commit is contained in:
parent
1e219fef0a
commit
b0cff949f5
|
@ -1,15 +1,10 @@
|
||||||
import unittest
|
import unittest
|
||||||
|
|
||||||
from TTS.tts.utils.text.characters import BaseCharacters, Graphemes, IPAPhonemes, create_graphemes, create_phonemes
|
from TTS.tts.utils.text.characters import BaseCharacters, Graphemes, IPAPhonemes
|
||||||
|
|
||||||
# pylint: disable=protected-access
|
# pylint: disable=protected-access
|
||||||
|
|
||||||
|
|
||||||
def test_make_symbols():
|
|
||||||
_ = create_phonemes()
|
|
||||||
_ = create_graphemes()
|
|
||||||
|
|
||||||
|
|
||||||
class BaseCharacterTest(unittest.TestCase):
|
class BaseCharacterTest(unittest.TestCase):
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
self.characters_empty = BaseCharacters("", "", pad="", eos="", bos="", blank="", is_unique=True, is_sorted=True)
|
self.characters_empty = BaseCharacters("", "", pad="", eos="", bos="", blank="", is_unique=True, is_sorted=True)
|
||||||
|
|
|
@ -64,6 +64,7 @@ class TestTTSTokenizer(unittest.TestCase):
|
||||||
def test_init_from_config(self):
|
def test_init_from_config(self):
|
||||||
@dataclass
|
@dataclass
|
||||||
class Characters(Coqpit):
|
class Characters(Coqpit):
|
||||||
|
characters_class: str = None
|
||||||
characters: str = _phonemes
|
characters: str = _phonemes
|
||||||
punctuations: str = _punctuations
|
punctuations: str = _punctuations
|
||||||
pad: str = _pad
|
pad: str = _pad
|
||||||
|
|
|
@ -11,7 +11,7 @@ from TTS.tts.layers.losses import GlowTTSLoss
|
||||||
from TTS.tts.models.glow_tts import GlowTTS
|
from TTS.tts.models.glow_tts import GlowTTS
|
||||||
from TTS.tts.utils.speakers import SpeakerManager
|
from TTS.tts.utils.speakers import SpeakerManager
|
||||||
from TTS.utils.audio import AudioProcessor
|
from TTS.utils.audio import AudioProcessor
|
||||||
from TTS.utils.logging.tensorboard_logger import TensorboardLogger
|
from trainer.logging.tensorboard_logger import TensorboardLogger
|
||||||
|
|
||||||
# pylint: disable=unused-variable
|
# pylint: disable=unused-variable
|
||||||
|
|
||||||
|
|
|
@ -20,6 +20,7 @@ config = MultibandMelganConfig(
|
||||||
eval_split_size=1,
|
eval_split_size=1,
|
||||||
print_step=1,
|
print_step=1,
|
||||||
print_eval=True,
|
print_eval=True,
|
||||||
|
steps_to_start_discriminator=1,
|
||||||
data_path="tests/data/ljspeech",
|
data_path="tests/data/ljspeech",
|
||||||
discriminator_model_params={"base_channels": 16, "max_channels": 64, "downsample_factors": [4, 4, 4]},
|
discriminator_model_params={"base_channels": 16, "max_channels": 64, "downsample_factors": [4, 4, 4]},
|
||||||
output_path=output_path,
|
output_path=output_path,
|
||||||
|
|
Loading…
Reference in New Issue