From 91e4a1b316e03ee5c93c1ae0f7cc148d6ed6d91f Mon Sep 17 00:00:00 2001 From: julienChemillier Date: Wed, 8 Mar 2023 20:59:53 +0100 Subject: [PATCH] Fix creation.c --- src/cnn/creation.c | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/cnn/creation.c b/src/cnn/creation.c index 12f32f4..6b0402f 100644 --- a/src/cnn/creation.c +++ b/src/cnn/creation.c @@ -235,7 +235,7 @@ void add_dense(Network* network, int size_output, int activation) { } } - initialisation_1d_matrix(network->initialisation, nn->bias, size_output, size_input, size_output); + initialisation_1d_matrix(network->initialisation, nn->bias, size_output, size_input); initialisation_2d_matrix(network->initialisation, nn->weights, size_input, size_output, size_input, size_output); create_a_line_input_layer(network, n, size_output); create_a_line_input_z_layer(network, n, size_output); @@ -275,7 +275,7 @@ void add_dense_linearisation(Network* network, int size_output, int activation) nn->d_weights[i][j] = 0.; } } - initialisation_1d_matrix(network->initialisation, nn->bias, size_output, size_input, size_output); + initialisation_1d_matrix(network->initialisation, nn->bias, size_output, size_input); initialisation_2d_matrix(network->initialisation, nn->weights, size_input, size_output, size_input, size_output); create_a_line_input_layer(network, n, size_output); create_a_line_input_z_layer(network, n, size_output);