diff --git a/dnn/training_tf2/lpcnet.py b/dnn/training_tf2/lpcnet.py index ffb239043a67b2063981d840fb3b3955424147dc..46e66bcef9534425fa26475548f160ca9c715f70 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: