Update comment and add a warning

This commit is contained in:
Eren Gölge 2021-09-07 12:23:32 +00:00
parent 674c72b1f3
commit e20ea57c87
2 changed files with 2 additions and 2 deletions

View File

@ -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

View File

@ -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