def make_dataloader(opt):
# make train's dataloader
train_mix_reader = AudioData(opt['datasets']['train']['dataroot_mix'],
**opt['datasets']['audio_setting'])
train_target_readers = [
AudioData(opt['datasets']['train']['dataroot_targets'][0],
**opt['datasets']['audio_setting']),
AudioData(opt['datasets']['train']['dataroot_targets'][1],
**opt['datasets']['audio_setting'])
]
train_dataset = Dataloader.dataset(train_mix_reader, train_target_readers)
train_dataloader = Dataloader.dataloader(
train_dataset, **opt['datasets']['dataloader_setting'])
# make validation dataloader
val_mix_reader = AudioData(opt['datasets']['val']['dataroot_mix'],
**opt['datasets']['audio_setting'])
val_target_readers = [
AudioData(opt['datasets']['val']['dataroot_targets'][0],
**opt['datasets']['audio_setting']),
AudioData(opt['datasets']['val']['dataroot_targets'][1],
**opt['datasets']['audio_setting'])
]
val_dataset = Dataloader.dataset(val_mix_reader, val_target_readers)
val_dataloader = Dataloader.dataloader(
val_dataset, **opt['datasets']['dataloader_setting'])
return train_dataloader, val_dataloader
def make_dataloader(opt):
# make train's dataloader
train_mix_reader = AudioData(opt['datasets']['train']['dataroot_mix'],
**opt['datasets']['audio_setting'])
train_target_readers = [
AudioData(opt['datasets']['train']['dataroot_targets'][0],
**opt['datasets']['audio_setting']),
AudioData(opt['datasets']['train']['dataroot_targets'][1],
**opt['datasets']['audio_setting'])
]
train_dataset = Dataloader.dataset(
train_mix_reader, train_target_readers,
opt['datasets']['dataloader_setting']['cmvn_file'])
train_dataloader = Loader(
train_dataset,
batch_size=opt['datasets']['dataloader_setting']['batch_size'],
num_workers=opt['datasets']['dataloader_setting']['num_workers'],
shuffle=opt['datasets']['dataloader_setting']['shuffle'])
# make validation dataloader
val_mix_reader = AudioData(opt['datasets']['val']['dataroot_mix'],
**opt['datasets']['audio_setting'])
val_target_readers = [
AudioData(opt['datasets']['val']['dataroot_targets'][0],
**opt['datasets']['audio_setting']),
AudioData(opt['datasets']['val']['dataroot_targets'][1],
**opt['datasets']['audio_setting'])
]
val_dataset = Dataloader.dataset(
val_mix_reader, val_target_readers,
opt['datasets']['dataloader_setting']['cmvn_file'])
val_dataloader = Loader(
val_dataset,
batch_size=opt['datasets']['dataloader_setting']['batch_size'],
num_workers=opt['datasets']['dataloader_setting']['num_workers'],
shuffle=opt['datasets']['dataloader_setting']['shuffle'])
return train_dataloader, val_dataloader
