Fix conv1d, default to size 384

This commit is contained in:
Jean-Marc Valin 2018-11-24 15:30:17 -05:00
parent 66486004ba
commit c025744e34
2 changed files with 2 additions and 4 deletions

View file

@ -115,7 +115,7 @@ def dump_embedding_layer(self, f, hf):
Embedding.dump_layer = dump_embedding_layer
model, _, _ = lpcnet.new_lpcnet_model(rnn_units1=640, use_gpu=False)
model, _, _ = lpcnet.new_lpcnet_model(rnn_units1=384, use_gpu=False)
model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['sparse_categorical_accuracy'])
#model.summary()

View file

@ -210,11 +210,9 @@ void compute_conv1d(const Conv1DLayer *layer, float *output, float *mem, const f
int stride;
float tmp[MAX_CONV_INPUTS];
celt_assert(layer->nb_inputs*layer->kernel_size <= MAX_CONV_INPUTS);
M = layer->nb_inputs;
N = layer->nb_neurons;
RNN_COPY(tmp, mem, layer->nb_inputs*(layer->kernel_size-1));
RNN_COPY(tmp, input, layer->nb_inputs);
M = layer->nb_inputs;
M = layer->nb_inputs*layer->kernel_size;
N = layer->nb_neurons;
stride = N;
for (i=0;i<N;i++)