leo4.0 / spec_gen.py
chilge's picture
Duplicate from innnky/sovits4
d18bd24
raw
history blame
560 Bytes
from data_utils import TextAudioSpeakerLoader
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