delete reduntant module
This commit is contained in:
parent
556da7c3a0
commit
a2f6fbcca2
|
|
@ -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
|
|
||||||
Loading…
Reference in New Issue