Merge pull request #5 from eschmidbauer/main

train.py fix: RuntimeError: stft requires the return_complex
zhenye234 · Sep 27, 2023 · 2bec3ff · 2bec3ff
2 parents 4408428 + c0968a1
commit 2bec3ff
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hifi-gan/meldataset.py b/hifi-gan/meldataset.py
@@ -64,7 +64,7 @@ def mel_spectrogram(y, n_fft, num_mels, sampling_rate, hop_size, win_size, fmin,
     y = y.squeeze(1)
 
     spec = torch.stft(y, n_fft, hop_length=hop_size, win_length=win_size, window=hann_window[str(y.device)],
-                      center=center, pad_mode='reflect', normalized=False, onesided=True)
+                      center=center, pad_mode='reflect', normalized=False, onesided=True, return_complex=False)
 
     spec = torch.sqrt(spec.pow(2).sum(-1)+(1e-9))