From 2dcbc108bcd95689cdd59361e0c0f9fb9215f173 Mon Sep 17 00:00:00 2001 From: innnky Date: Thu, 8 Dec 2022 21:16:11 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E4=BA=86=E9=A2=84=E5=85=88?= =?UTF-8?q?=E7=94=9F=E6=88=90spec=E7=9A=84=E8=84=9A=E6=9C=AC?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- spec_gen.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) create mode 100644 spec_gen.py diff --git a/spec_gen.py b/spec_gen.py new file mode 100644 index 00000000..85ad3188 --- /dev/null +++ b/spec_gen.py @@ -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 \ No newline at end of file