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

merge dev

This commit is contained in:
shahules786 2022-10-28 13:10:36 +05:30
parent 8d8eaa80d5 aa52d1ed93
commit acb68c9855
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
enhancer/data/dataset.py
View File

@ -8,7 +8,7 @@ import numpy as np
import pytorch_lightning as pl import pytorch_lightning as pl
import torch import torch
import torch.nn.functional as F import torch.nn.functional as F
from torch.utils.data import DataLoader, Dataset from torch.utils.data import DataLoader, Dataset, RandomSampler
from torch_audiomentations import Compose from torch_audiomentations import Compose
from enhancer.data.fileprocessor import Fileprocessor from enhancer.data.fileprocessor import Fileprocessor
@ -221,11 +221,13 @@ class TaskDataset(pl.LightningDataModule):
return generator.manual_seed(seed) return generator.manual_seed(seed)
def train_dataloader(self): def train_dataloader(self):
dataset = TrainDataset(self)
sampler = RandomSampler(dataset, generator=self.generator)
return DataLoader( return DataLoader(
TrainDataset(self), dataset,
batch_size=self.batch_size, batch_size=self.batch_size,
num_workers=self.num_workers, num_workers=self.num_workers,
generator=self.generator, sampler=sampler,
collate_fn=self.train_collatefn, collate_fn=self.train_collatefn,
) )
@ -328,7 +330,6 @@ class EnhancerDataset(TaskDataset):
return self.prepare_segment(*self._test[idx]) return self.prepare_segment(*self._test[idx])
def prepare_segment(self, file_dict: dict, start_time: float): def prepare_segment(self, file_dict: dict, start_time: float):
print(file_dict["clean"].split("/")[-1])
clean_segment = self.audio( clean_segment = self.audio(
file_dict["clean"], offset=start_time, duration=self.duration file_dict["clean"], offset=start_time, duration=self.duration
) )