Merge pull request #1602 from yt605155624/fix_dtype

[TTS]fix dtype of window of stft
4 years ago · b6e976a860
parent 8938483529 5347dbad3f
commit b6e976a860
1 changed files with 2 additions and 2 deletions
--- a/paddlespeech/t2s/modules/losses.py
+++ b/paddlespeech/t2s/modules/losses.py
@ -489,7 +489,7 @@ def stft(x,
    """
    # calculate window
    window = signal.get_window(window, win_length, fftbins=True)
-    window = paddle.to_tensor(window)
+    window = paddle.to_tensor(window, dtype=x.dtype)
    x_stft = paddle.signal.stft(
        x,
        fft_size,
@ -896,7 +896,7 @@ class MelSpectrogram(nn.Layer):
            # calculate window
            window = signal.get_window(
                self.window, self.win_length, fftbins=True)
-            window = paddle.to_tensor(window)
+            window = paddle.to_tensor(window, dtype=x.dtype)
        else:
            window = None