testadmin
/
ParakeetRebeccaRosario
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

merge wavenet

This commit is contained in:
chenfeiyu 2021-04-19 20:09:01 +08:00
parent 3741cc49ca 6749ce40ea
commit 9da118e53b
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
parakeet/modules/audio.py
View File

@ -71,6 +71,8 @@ def dequantize(quantized, n_bands, dtype=None):
value = (paddle.cast(quantized, dtype) + 0.5) * (2.0 / n_bands) - 1.0 value = (paddle.cast(quantized, dtype) + 0.5) * (2.0 / n_bands) - 1.0
return value return value
import librosa
librosa.stft(
class STFT(nn.Layer): class STFT(nn.Layer):
"""A module for computing stft transformation in a differentiable way. """A module for computing stft transformation in a differentiable way.
@ -117,7 +119,11 @@ class STFT(nn.Layer):
""" """
def __init__(self, n_fft, hop_length=None, win_length=None, window="hanning", center=True, pad_mode="reflect"): def __init__(self, n_fft, hop_length=None, win_length=None, window="hanning", center=True, pad_mode="reflect"):
<<<<<<< HEAD
super().__init__() super().__init__()
=======
super(STFT, self).__init__()
>>>>>>> 6749ce40eab7c5de0ca11fe5d2fd57120b10415c
# By default, use the entire frame # By default, use the entire frame
if win_length is None: if win_length is None:
win_length = n_fft win_length = n_fft