Skip to content

Commit

Permalink
添加了预先生成spec的脚本
Browse files Browse the repository at this point in the history
  • Loading branch information
innnky committed Dec 8, 2022
1 parent fce79df commit 2dcbc10
Showing 1 changed file with 22 additions and 0 deletions.
22 changes: 22 additions & 0 deletions spec_gen.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,22 @@
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

0 comments on commit 2dcbc10

Please sign in to comment.