File tree Expand file tree Collapse file tree 1 file changed +2
-0
lines changed Expand file tree Collapse file tree 1 file changed +2
-0
lines changed Original file line number Diff line number Diff 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); \
You can’t perform that action at this time.
0 commit comments