Skip to content

Commit 3fe3790

Browse files
committed
Fix to RNN file reader to set _size correctly.
1 parent 42e1ffb commit 3fe3790

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/rnn_reader.c

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -111,6 +111,7 @@ RNNModel *rnnoise_model_from_file(FILE *f)
111111
#define INPUT_DENSE(name) do { \
112112
INPUT_VAL(name->nb_inputs); \
113113
INPUT_VAL(name->nb_neurons); \
114+
ret->name ## _size = name->nb_neurons; \
114115
INPUT_ACTIVATION(name->activation); \
115116
INPUT_ARRAY(name->input_weights, name->nb_inputs * name->nb_neurons); \
116117
INPUT_ARRAY(name->bias, name->nb_neurons); \
@@ -119,6 +120,7 @@ RNNModel *rnnoise_model_from_file(FILE *f)
119120
#define INPUT_GRU(name) do { \
120121
INPUT_VAL(name->nb_inputs); \
121122
INPUT_VAL(name->nb_neurons); \
123+
ret->name ## _size = name->nb_neurons; \
122124
INPUT_ACTIVATION(name->activation); \
123125
INPUT_ARRAY(name->input_weights, name->nb_inputs * name->nb_neurons * 3); \
124126
INPUT_ARRAY(name->recurrent_weights, name->nb_neurons * name->nb_neurons * 3); \

0 commit comments

Comments
 (0)