From 22eb9256e20ff401f14fc79c25bf19c47dad026f Mon Sep 17 00:00:00 2001 From: shahules786 Date: Mon, 12 Sep 2022 11:33:15 +0530 Subject: [PATCH] fix typo --- enhancer/data/dataset.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/enhancer/data/dataset.py b/enhancer/data/dataset.py index aa98219..797d691 100644 --- a/enhancer/data/dataset.py +++ b/enhancer/data/dataset.py @@ -81,13 +81,13 @@ class TaskDataset(pl.LightningDataModule): num_segments = round(total_dur/self.duration) for index in range(num_segments): start_time = index * self.duration - self._validation.append(({"clean_file":clean,"noisy_file":noisy}, + self._validation.append(({"clean":clean,"noisy":noisy}, start_time)) def train_dataloader(self): - return DataLoader(TrainDataset(self), batch_size = self.batch_size) + return DataLoader(TrainDataset(self), batch_size = self.batch_size,num_workers=2) def val_dataloader(self): - return DataLoader(ValidDataset(self), batch_size = self.batch_size) + return DataLoader(ValidDataset(self), batch_size = self.batch_size,num_workers=2) class EnhancerDataset(TaskDataset): """Dataset object for creating clean-noisy speech enhancement datasets"""