mel2samp.py: adding fmin and fmax params

deterministic-algorithms-lab · Nov 9, 2018 · 63bdb3f · 63bdb3f
1 parent 4ef2bb9
commit 63bdb3f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/mel2samp.py b/mel2samp.py
@@ -63,14 +63,15 @@ class Mel2Samp(torch.utils.data.Dataset):
     spectrogram, audio pair.
     """
     def __init__(self, training_files, segment_length, filter_length,
-                 hop_length, win_length, sampling_rate):
+                 hop_length, win_length, sampling_rate, mel_fmin, mel_fmax):
         self.audio_files = files_to_list(training_files)
         random.seed(1234)
         random.shuffle(self.audio_files)
         self.stft = TacotronSTFT(filter_length=filter_length,
                                  hop_length=hop_length,
                                  win_length=win_length,
-                                 sampling_rate=sampling_rate)
+                                 sampling_rate=sampling_rate,
+                                 mel_fmin=mel_fmin, mel_fmax=mel_fmax)
         self.segment_length = segment_length
         self.sampling_rate = sampling_rate