Training seems to work

This commit is contained in:
Jean-Marc Valin 2018-08-16 22:43:13 -04:00
parent 785a2b2e84
commit 87cd75f6f4
2 changed files with 14 additions and 4 deletions

View file

@ -41,7 +41,7 @@ class PCMInit(Initializer):
}
def new_wavernn_model():
pcm = Input(shape=(None, 1))
pcm = Input(shape=(None, 2))
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))(embed(pcm))
cpcm = Reshape((-1, 128*2))(embed(pcm))
cfeat = fconv2(fconv1(feat))