From aef9c9bff64f3a66e7db1ec3e8820132f9c3be3e Mon Sep 17 00:00:00 2001 From: Jiko Date: Fri, 30 May 2025 11:32:19 +0800 Subject: [PATCH 1/2] =?UTF-8?q?=E8=A7=A3=E5=86=B3=E5=86=85=E5=AD=98?= =?UTF-8?q?=E6=B3=84=E9=9C=B2=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- speakerlab/dataset/dataset.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/speakerlab/dataset/dataset.py b/speakerlab/dataset/dataset.py index 97281ecaf..97edffc4f 100644 --- a/speakerlab/dataset/dataset.py +++ b/speakerlab/dataset/dataset.py @@ -24,12 +24,11 @@ def __getitem__(self, index): spkid = self.preprocessor['label_encoder'](spk, speed_index) wav = self.preprocessor['augmentations'](wav) feat = self.preprocessor['feature_extractor'](wav) + self.data_keys = list(self.data_points.keys()) return feat, spkid def get_data(self, index): - if not hasattr(self, 'data_keys'): - self.data_keys = list(self.data_points.keys()) key = self.data_keys[index] return self.data_points[key] From 5f259c5b9db9f19309d91d364ced6dfca683a2c7 Mon Sep 17 00:00:00 2001 From: Jiko Date: Fri, 30 May 2025 11:37:05 +0800 Subject: [PATCH 2/2] =?UTF-8?q?=E5=86=85=E5=AD=98=E6=B3=84=E9=9C=B2?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- speakerlab/dataset/dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/speakerlab/dataset/dataset.py b/speakerlab/dataset/dataset.py index 97edffc4f..b726c9b72 100644 --- a/speakerlab/dataset/dataset.py +++ b/speakerlab/dataset/dataset.py @@ -9,6 +9,7 @@ class BaseSVDataset(Dataset): def __init__(self, data_file: str, preprocessor: dict): self.data_points = self.read_file(data_file) self.preprocessor = preprocessor + self.data_keys = list(self.data_points.keys()) def __len__(self): return len(self.data_points) @@ -24,7 +25,6 @@ def __getitem__(self, index): spkid = self.preprocessor['label_encoder'](spk, speed_index) wav = self.preprocessor['augmentations'](wav) feat = self.preprocessor['feature_extractor'](wav) - self.data_keys = list(self.data_points.keys()) return feat, spkid