Skip to content

Commit a715054

Browse files
committed
More neural network models
1 parent 407204d commit a715054

File tree

5 files changed

+22103
-3
lines changed

5 files changed

+22103
-3
lines changed

Makefile.am

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,8 @@ librnnoise_la_SOURCES = \
2424
src/rnn_data.c \
2525
src/models/cb.c \
2626
src/models/mp.c \
27+
src/models/bd.c \
28+
src/models/lq.c \
2729
src/rnn_reader.c \
2830
src/models.c \
2931
src/pitch.c \

TRAINING

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,4 +8,4 @@
88

99
(4) ./rnn_train.py
1010

11-
(5) ./dump_rnn.py weights.hdf5 ../src/rnn_data.c ../src/rnn_data.h
11+
(5) ./dump_rnn.py weights.hdf5 rnn_data.c rnn_data.rnnn name

src/models.c

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -39,18 +39,24 @@ static const char *model_names[] = {
3939
"orig",
4040
"cb",
4141
"mp",
42+
"bd",
43+
"lq",
4244
NULL
4345
};
4446

4547
extern const struct RNNModel
4648
model_orig,
4749
model_cb,
48-
model_mp;
50+
model_mp,
51+
model_bd,
52+
model_lq;
4953

5054
static const struct RNNModel *models[] = {
5155
&model_orig,
5256
&model_cb,
53-
&model_mp
57+
&model_mp,
58+
&model_bd,
59+
&model_lq
5460
};
5561

5662
const char **rnnoise_models()

0 commit comments

Comments
 (0)