From a2f6fbcca29c7d1690ddec5d39afe67db555779b Mon Sep 17 00:00:00 2001 From: shahules786 Date: Wed, 31 Aug 2022 11:04:36 +0530 Subject: [PATCH] delete reduntant module --- enhancer/data/dataset.py | 34 ---------------------------------- 1 file changed, 34 deletions(-) delete mode 100644 enhancer/data/dataset.py diff --git a/enhancer/data/dataset.py b/enhancer/data/dataset.py deleted file mode 100644 index 6b44001..0000000 --- a/enhancer/data/dataset.py +++ /dev/null @@ -1,34 +0,0 @@ -import os -import pytorch_lightning as pl -from typing import Optional - -from enhancer.data.vctk import Vctk -from enhancer.utils.config import Files - -DATASETS = ["Vctk"] - -class Dataset(pl.LightningDataModule): - - def __init__(self,name:str, directory:str, files:Files, - duration:float=1.0, sampling_rate:int=48000): - super().__init__() - - self.train_clean = os.path.join(directory,Files.train_clean) - self.train_noisy = os.path.join(directory,Files.train_noisy) - self.valid_clean = os.path.join(directory,Files.test_clean) - self.valid_noisy = os.path.join(directory,Files.test_noisy) - - if name.title() in DATASETS: - self.data_obj = eval(name.title) - - self.duration = duration - self.sampling_rate = sampling_rate - - def setup(self, stage: Optional[str] = None): - self.train_dataset = self.data_obj() - - def train_loader(self): - pass - - def valid_loader(self): - pass