From 659463e4b78ce0dbb044833d71485906ce061305 Mon Sep 17 00:00:00 2001 From: Julien Chemillier Date: Mon, 3 Oct 2022 10:06:31 +0200 Subject: [PATCH] Fix issue in lenet5 --- src/cnn/creation.c | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/cnn/creation.c b/src/cnn/creation.c index 732a34c..4e8ba10 100644 --- a/src/cnn/creation.c +++ b/src/cnn/creation.c @@ -33,7 +33,7 @@ Network* create_network_lenet5(int learning_rate, int dropout, int activation, i Network* network = create_network(8, learning_rate, dropout, initialisation, input_dim, input_depth); network->kernel[0]->activation = activation; network->kernel[0]->linearisation = 0; - add_convolution(network, 1, 32, 6, 28, activation); + add_convolution(network, input_depth, input_dim, 6, 28, activation); add_2d_average_pooling(network, 28, 14); add_convolution(network, 6, 14, 16, 10, activation); add_2d_average_pooling(network, 10, 5);