shithub: opus

Info  •  Files  •  Log  •  Branches

oops, fix number of columns

--- a/dnn/nnet.c

+++ b/dnn/nnet.c

@@ -303,7 +303,7 @@

       for (i=0;i<N;i++)

          recur[k*N + i] += gru->diag_weights[k*N + i]*state[i];

    }

-   sparse_sgemv_accum8x4(recur, gru->recurrent_weights, 3*N, 3*N, gru->idx, state);

+   sparse_sgemv_accum8x4(recur, gru->recurrent_weights, 3*N, N, gru->idx, state);

    for (i=0;i<2*N;i++)

       zrh[i] += recur[i];

    compute_activation(zrh, zrh, 2*N, ACTIVATION_SIGMOID);

-- 

⑨