mirror of https://github.com/coqui-ai/TTS.git
Improve data_path resolvement (#1567)
parent
2fc38f67d2
commit
f9d91a55f2
|
@ -6,7 +6,7 @@ import numpy as np
|
|||
import torch
|
||||
from torch.utils.data import DataLoader
|
||||
|
||||
from tests import get_tests_output_path
|
||||
from tests import get_tests_data_path, get_tests_output_path
|
||||
from TTS.tts.configs.shared_configs import BaseDatasetConfig, BaseTTSConfig
|
||||
from TTS.tts.datasets import TTSDataset, load_tts_samples
|
||||
from TTS.tts.utils.text.tokenizer import TTSTokenizer
|
||||
|
@ -20,7 +20,7 @@ os.makedirs(OUTPATH, exist_ok=True)
|
|||
# create a dummy config for testing data loaders.
|
||||
c = BaseTTSConfig(text_cleaner="english_cleaners", num_loader_workers=0, batch_size=2, use_noise_augment=False)
|
||||
c.r = 5
|
||||
c.data_path = "tests/data/ljspeech/"
|
||||
c.data_path = os.path.join(get_tests_data_path(), "ljspeech/")
|
||||
ok_ljspeech = os.path.exists(c.data_path)
|
||||
|
||||
dataset_config = BaseDatasetConfig(
|
||||
|
|
Loading…
Reference in New Issue