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

Biasing for overestimating the pitch correlation

Browse source
This commit is contained in:
Jean-Marc Valin 2022-02-25 13:55:21 -05:00
parent 05f02aaa49
commit e034b1096b
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
dnn/training_tf2/train_plc.py
View file

@ -104,7 +104,7 @@ def plc_loss(alpha=1.0, bias=0.):
e = (y_pred - y_true)*mask
e_bands = tf.signal.idct(e[:,:,:-2], norm='ortho')
bias_mask = K.minimum(1., K.maximum(0., 4*y_true[:,:,-1:]))
l1_loss = K.mean(K.abs(e)) + alpha*K.mean(K.abs(e_bands) + bias*bias_mask*K.maximum(0., e_bands)) + K.mean(K.minimum(K.abs(e[:,:,18:19]),1.)) + 8*K.mean(K.minimum(K.abs(e[:,:,18:19]),.4))
l1_loss = K.mean(K.abs(e)) + 0.1*K.mean(K.maximum(0., -e[:,:,-1:])) + alpha*K.mean(K.abs(e_bands) + bias*bias_mask*K.maximum(0., e_bands)) + K.mean(K.minimum(K.abs(e[:,:,18:19]),1.)) + 8*K.mean(K.minimum(K.abs(e[:,:,18:19]),.4))
return l1_loss
return loss