From e20ea57c87a0502cb250d7963f2f375ad8321755 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Tue, 7 Sep 2021 12:23:32 +0000 Subject: [PATCH] Update comment and add a warning --- TTS/trainer.py | 2 +- TTS/vocoder/datasets/preprocess.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/TTS/trainer.py b/TTS/trainer.py index bc9a49c6..13439861 100644 --- a/TTS/trainer.py +++ b/TTS/trainer.py @@ -205,7 +205,7 @@ class Trainer: # load data for `tts` models self.data_train, self.data_eval = load_meta_data(self.config.datasets) elif self.config.feature_path is not None: - # load data for `vocoder`models + # load pre-comnputed features for `vocoder`models print(f" > Loading features from: {self.config.feature_path}") self.data_eval, self.data_train = load_wav_feat_data( self.config.data_path, self.config.feature_path, self.config.eval_split_size diff --git a/TTS/vocoder/datasets/preprocess.py b/TTS/vocoder/datasets/preprocess.py index c4569b3d..62bd4ba5 100644 --- a/TTS/vocoder/datasets/preprocess.py +++ b/TTS/vocoder/datasets/preprocess.py @@ -59,7 +59,7 @@ def load_wav_feat_data(data_path, feat_path, eval_split_size): wav_paths.sort(key=lambda x: Path(x).stem) feat_paths.sort(key=lambda x: Path(x).stem) - assert len(wav_paths) == len(feat_paths) + assert len(wav_paths) == len(feat_paths), f" [!] {len(wav_paths)} vs {feat_paths}" for wav, feat in zip(wav_paths, feat_paths): wav_name = Path(wav).stem feat_name = Path(feat).stem