Combine LAR+L1 regularization

This commit is contained in:
Jean-Marc Valin 2021-10-13 02:33:35 -04:00
parent 054d984bf3
commit fe7b54c0e8
4 changed files with 33 additions and 8 deletions

View file

@ -304,8 +304,8 @@ def new_lpcnet_model(rnn_units1=384, rnn_units2=16, nb_used_features=20, batch_s
if not flag_e2e:
model = Model([pcm, feat, pitch], ulaw_prob)
else:
m_out = Concatenate()([tensor_preds,ulaw_prob])
model = Model([pcm, feat, pitch], m_out)
m_out = Concatenate(name='pdf')([tensor_preds,ulaw_prob])
model = Model([pcm, feat, pitch], [m_out, cfeat])
model.rnn_units1 = rnn_units1
model.rnn_units2 = rnn_units2
model.nb_used_features = nb_used_features