Predicting pre-emphasized audio

This commit is contained in:
Jean-Marc Valin 2018-08-16 13:58:33 -04:00
parent 3d20cdaed4
commit 785a2b2e84
3 changed files with 11 additions and 11 deletions

View file

@ -10,7 +10,7 @@ import numpy as np
import h5py
import sys
rnn_units=64
rnn_units=512
pcm_bits = 8
pcm_levels = 2**pcm_bits
nb_used_features = 38
@ -41,7 +41,7 @@ class PCMInit(Initializer):
}
def new_wavernn_model():
pcm = Input(shape=(None, 2))
pcm = Input(shape=(None, 1))
pitch = Input(shape=(None, 1))
feat = Input(shape=(None, nb_used_features))
dec_feat = Input(shape=(None, 32))
@ -61,7 +61,7 @@ def new_wavernn_model():
cpitch = pitch
embed = Embedding(256, 128, embeddings_initializer=PCMInit())
cpcm = Reshape((-1, 128*2))(embed(pcm))
cpcm = Reshape((-1, 128))(embed(pcm))
cfeat = fconv2(fconv1(feat))