From cb05ca297925f24c0dfda3c838f914a474015601 Mon Sep 17 00:00:00 2001 From: megemini Date: Wed, 11 Dec 2024 14:20:57 +0800 Subject: [PATCH] [Fix] empty npy folder --- .../vector/exps/ge2e/speaker_verification_dataset.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/paddlespeech/vector/exps/ge2e/speaker_verification_dataset.py b/paddlespeech/vector/exps/ge2e/speaker_verification_dataset.py index 87b7d59a3..873ebfeef 100644 --- a/paddlespeech/vector/exps/ge2e/speaker_verification_dataset.py +++ b/paddlespeech/vector/exps/ge2e/speaker_verification_dataset.py @@ -39,7 +39,10 @@ class MultiSpeakerMelDataset(Dataset): def __init__(self, dataset_root: Path): self.root = Path(dataset_root).expanduser() - speaker_dirs = [f for f in self.root.glob("*") if f.is_dir()] + speaker_dirs = [ + f for f in self.root.glob("*") + if f.is_dir() and list(f.glob("*.npy")) + ] speaker_utterances = { speaker_dir: list(speaker_dir.glob("*.npy"))