| from data_utils import TextAudioSpeakerLoader, EvalDataLoader |
| import json |
| from tqdm import tqdm |
|
|
| from utils import HParams |
|
|
| config_path = 'configs/config.json' |
| with open(config_path, "r") as f: |
| data = f.read() |
| config = json.loads(data) |
| hps = HParams(**config) |
|
|
| train_dataset = TextAudioSpeakerLoader("filelists/train.txt", hps) |
| test_dataset = TextAudioSpeakerLoader("filelists/test.txt", hps) |
| eval_dataset = TextAudioSpeakerLoader("filelists/val.txt", hps) |
|
|
| for _ in tqdm(train_dataset): |
| pass |
| for _ in tqdm(eval_dataset): |
| pass |
| for _ in tqdm(test_dataset): |
| pass |