Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix typo

This commit is contained in:
shahules786 2022-09-12 11:33:15 +05:30
parent 0e72bb3bb8
commit 22eb9256e2
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
enhancer/data/dataset.py
View File

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