def get_dataset(config, model): if config['dataset']['type'] == 'vctk+demand': return datasets.VCTKAndDEMANDDataset(config, model).load_dataset() #just nsdtsea dataset is used here elif config['dataset']['type'] == 'nsdtsea': return datasets.NSDTSEADataset(config, model).load_dataset()
def get_dataset(config, model): if config['dataset']['type'] == 'vctk+demand': return datasets.VCTKAndDEMANDDataset(config, model).load_dataset() elif config['dataset']['type'] == 'nsdtsea': return datasets.NSDTSEADataset(config, model).load_dataset() elif config['dataset']['type'] == 'wsj0-mix': return datasets.WSJ0(config, model).load_dataset()