eden-emu/opus
15
0
Fork 0
mirror of https://github.com/xiph/opus.git synced 2025-05-19 18:08:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

...

Browse source
This commit is contained in:
Jean-Marc Valin 2018-10-10 16:48:51 -04:00
parent 495f8ea5f3
commit 9756feefbd
2 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
dnn/lpcnet.py
View file

@ -10,7 +10,7 @@ import numpy as np
import h5py
import sys
rnn_units1=128
rnn_units1=256
rnn_units2=32
pcm_bits = 8
embed_size = 128
@ -77,7 +77,7 @@ def new_wavernn_model():
rnn_in = Concatenate()([cpcm, cexc, rep(cfeat)])
md = MDense(pcm_levels, activation='softmax')
gru_out1, _ = rnn(rnn_in)
gru_out2, _ = rnn2(gru_out1)
gru_out2, _ = rnn2(Concatenate()([gru_out1, rep(cfeat)]))
ulaw_prob = md(gru_out2)
model = Model([pcm, exc, feat, pitch], ulaw_prob)
@ -85,7 +85,7 @@ def new_wavernn_model():
dec_rnn_in = Concatenate()([cpcm, cexc, dec_feat])
dec_gru_out1, state1 = rnn(dec_rnn_in, initial_state=dec_state1)
dec_gru_out2, state2 = rnn2(dec_gru_out1, initial_state=dec_state2)
dec_gru_out2, state2 = rnn2(Concatenate()([dec_gru_out1, dec_feat]), initial_state=dec_state2)
dec_ulaw_prob = md(dec_gru_out2)
decoder = Model([pcm, exc, dec_feat, dec_state1, dec_state2], [dec_ulaw_prob, state1, state2])