eden-emu/opus
15
0
Fork 0
mirror of https://github.com/xiph/opus.git synced 2025-05-25 20:59:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Bring test_lpcnet.py up-to-date

Browse source
This commit is contained in:
Jean-Marc Valin 2021-10-20 04:01:12 -04:00
parent 144b7311bc
commit 3f7756c53f
2 changed files with 16 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

4
dnn/training_tf2/lpcnet.py
View file

@ -313,7 +313,7 @@ def new_lpcnet_model(rnn_units1=384, rnn_units2=16, nb_used_features=20, batch_s
if not flag_e2e:
encoder = Model([feat, pitch], cfeat)
dec_rnn_in = Concatenate()([cpcm, dec_feat])
dec_rnn_in = Concatenate()([cpcm_decoder, dec_feat])
else:
encoder = Model([feat, pitch], [cfeat,lpcoeffs])
dec_rnn_in = Concatenate()([cpcm_decoder, dec_feat])
@ -324,5 +324,5 @@ def new_lpcnet_model(rnn_units1=384, rnn_units2=16, nb_used_features=20, batch_s
if flag_e2e:
decoder = Model([dpcm, dec_feat, dec_state1, dec_state2], [dec_ulaw_prob, state1, state2])
else:
decoder = Model([pcm, dec_feat, dec_state1, dec_state2, lpcoeffs], [dec_ulaw_prob, state1, state2])
decoder = Model([dpcm, dec_feat, dec_state1, dec_state2], [dec_ulaw_prob, state1, state2])
return model, encoder, decoder