shithub: opus

Info  •  Files  •  Log  •  Branches

Freeze LPCs when quantizing e2e models

--- a/dnn/training_tf2/lpcnet.py

+++ b/dnn/training_tf2/lpcnet.py

@@ -249,6 +249,12 @@

     fdense1 = Dense(128, activation='tanh', name='feature_dense1')

     fdense2 = Dense(128, activation='tanh', name='feature_dense2')

+    if flag_e2e and quantize:

+        fconv1.trainable = False

+        fconv2.trainable = False

+        fdense1.trainable = False

+        fdense2.trainable = False

+

     cfeat = fdense2(fdense1(cfeat))

     if not flag_e2e:

-- 

⑨