Fix to RNN file reader to set _size correctly.

xiph · Apr 25, 2019 · 3fe3790 · 3fe3790
1 parent 42e1ffb
commit 3fe3790
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/rnn_reader.c b/src/rnn_reader.c
@@ -111,6 +111,7 @@ RNNModel *rnnoise_model_from_file(FILE *f)
 #define INPUT_DENSE(name) do { \
     INPUT_VAL(name->nb_inputs); \
     INPUT_VAL(name->nb_neurons); \
+    ret->name ## _size = name->nb_neurons; \
     INPUT_ACTIVATION(name->activation); \
     INPUT_ARRAY(name->input_weights, name->nb_inputs * name->nb_neurons); \
     INPUT_ARRAY(name->bias, name->nb_neurons); \
@@ -119,6 +120,7 @@ RNNModel *rnnoise_model_from_file(FILE *f)
 #define INPUT_GRU(name) do { \
     INPUT_VAL(name->nb_inputs); \
     INPUT_VAL(name->nb_neurons); \
+    ret->name ## _size = name->nb_neurons; \
     INPUT_ACTIVATION(name->activation); \
     INPUT_ARRAY(name->input_weights, name->nb_inputs * name->nb_neurons * 3); \
     INPUT_ARRAY(name->recurrent_weights, name->nb_neurons * name->nb_neurons * 3); \