fix(vol_aug)

This commit is contained in:
ylzz1997 2023-05-29 01:22:57 +08:00
parent 32337a0c5b
commit e3dc2b2b94
1 changed files with 2 additions and 2 deletions

View File

@ -23,7 +23,7 @@ class TextAudioSpeakerLoader(torch.utils.data.Dataset):
3) computes spectrograms from audio files.
"""
def __init__(self, audiopaths, hparams, all_in_mem: bool = False, vol_aug: bool = False):
def __init__(self, audiopaths, hparams, all_in_mem: bool = False):
self.audiopaths = load_filepaths_and_text(audiopaths)
self.hparams = hparams
self.max_wav_value = hparams.data.max_wav_value
@ -36,7 +36,7 @@ class TextAudioSpeakerLoader(torch.utils.data.Dataset):
self.spec_len = hparams.train.max_speclen
self.spk_map = hparams.spk
self.vol_emb = hparams.model.vol_embedding
self.vol_aug = vol_aug
self.vol_aug = hparams.train.vol_aug
random.seed(1234)
random.shuffle(self.audiopaths)