From 1f5189a50a41a07ebc15f0d3423ef236de9f9893 Mon Sep 17 00:00:00 2001
From: Jean-Marc Valin <jmvalin@jmvalin.ca>
Date: Sat, 2 Mar 2024 14:38:34 -0500
Subject: [PATCH] Actually use RTCD in the PLC feature prediction
---
dnn/lpcnet_plc.c | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/dnn/lpcnet_plc.c b/dnn/lpcnet_plc.c
index 2aecb859a..2a901ad27 100644
--- a/dnn/lpcnet_plc.c
+++ b/dnn/lpcnet_plc.c
@@ -112,10 +112,10 @@ static void compute_plc_pred(LPCNetPLCState *st, float *out, const float *in) {
PLCModel *model = &st->model;
PLCNetState *net = &st->plc_net;
celt_assert(st->loaded);
- compute_generic_dense(&model->plc_dense_in, tmp, in, ACTIVATION_TANH, 0);
- compute_generic_gru(&model->plc_gru1_input, &model->plc_gru1_recurrent, net->gru1_state, tmp, 0);
- compute_generic_gru(&model->plc_gru2_input, &model->plc_gru2_recurrent, net->gru2_state, net->gru1_state, 0);
- compute_generic_dense(&model->plc_dense_out, out, net->gru2_state, ACTIVATION_LINEAR, 0);
+ compute_generic_dense(&model->plc_dense_in, tmp, in, ACTIVATION_TANH, st->arch);
+ compute_generic_gru(&model->plc_gru1_input, &model->plc_gru1_recurrent, net->gru1_state, tmp, st->arch);
+ compute_generic_gru(&model->plc_gru2_input, &model->plc_gru2_recurrent, net->gru2_state, net->gru1_state, st->arch);
+ compute_generic_dense(&model->plc_dense_out, out, net->gru2_state, ACTIVATION_LINEAR, st->arch);
}
static int get_fec_or_pred(LPCNetPLCState *st, float *out) {
--
GitLab