From ae9d38aa3109d6e099c0be1e36be6b78d758db1f Mon Sep 17 00:00:00 2001 From: mrq Date: Fri, 18 Aug 2023 23:57:07 -0500 Subject: [PATCH] forgot to have it pull from specified noise to the hdf5 dataset --- vall_e/config.py | 3 +-- vall_e/data.py | 4 ++++ 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/vall_e/config.py b/vall_e/config.py index 958361e..3ff8a66 100755 --- a/vall_e/config.py +++ b/vall_e/config.py @@ -475,8 +475,7 @@ try: if not cfg.dataset.use_hdf5: cfg.dataset.training = [ Path(dir) for dir in cfg.dataset.training ] cfg.dataset.validation = [ Path(dir) for dir in cfg.dataset.validation ] - - cfg.dataset.noise = [ Path(dir) for dir in cfg.dataset.noise ] + cfg.dataset.noise = [ Path(dir) for dir in cfg.dataset.noise ] except Exception as e: pass diff --git a/vall_e/data.py b/vall_e/data.py index d746e2f..e455723 100755 --- a/vall_e/data.py +++ b/vall_e/data.py @@ -638,6 +638,10 @@ def create_dataset_hdf5(): for data_dir in tqdm(cfg.dataset.validation, desc='Processing Validation'): add( data_dir, type="validation" ) + # noise + for data_dir in tqdm(cfg.dataset.noise, desc='Processing Noise'): + add( data_dir, type="noise" ) + # write symmap hf.create_dataset('symmap', data=json.dumps(symmap))