From 054d984bf372c9c9f63fad2eb2e3751a25cac0a9 Mon Sep 17 00:00:00 2001
From: Jean-Marc Valin <jmvalin@amazon.com>
Date: Sat, 16 Oct 2021 17:20:24 -0400
Subject: [PATCH] Freeze LPCs when quantizing e2e models
---
dnn/training_tf2/lpcnet.py | 6 ++++++
1 file changed, 6 insertions(+)
diff --git a/dnn/training_tf2/lpcnet.py b/dnn/training_tf2/lpcnet.py
index ffb239043..46e66bcef 100644
--- a/dnn/training_tf2/lpcnet.py
+++ b/dnn/training_tf2/lpcnet.py
@@ -249,6 +249,12 @@ def new_lpcnet_model(rnn_units1=384, rnn_units2=16, nb_used_features=20, batch_s
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:
--
GitLab