Skip to content

Commit 63bdb3f

Browse files
committed
mel2samp.py: adding fmin and fmax params
1 parent 4ef2bb9 commit 63bdb3f

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

mel2samp.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -63,14 +63,15 @@ class Mel2Samp(torch.utils.data.Dataset):
6363
spectrogram, audio pair.
6464
"""
6565
def __init__(self, training_files, segment_length, filter_length,
66-
hop_length, win_length, sampling_rate):
66+
hop_length, win_length, sampling_rate, mel_fmin, mel_fmax):
6767
self.audio_files = files_to_list(training_files)
6868
random.seed(1234)
6969
random.shuffle(self.audio_files)
7070
self.stft = TacotronSTFT(filter_length=filter_length,
7171
hop_length=hop_length,
7272
win_length=win_length,
73-
sampling_rate=sampling_rate)
73+
sampling_rate=sampling_rate,
74+
mel_fmin=mel_fmin, mel_fmax=mel_fmax)
7475
self.segment_length = segment_length
7576
self.sampling_rate = sampling_rate
7677

0 commit comments

Comments
 (0)