From 95ce12358758a0c54b3d6e661777c7c266b816f5 Mon Sep 17 00:00:00 2001 From: augustin64 Date: Wed, 1 Mar 2023 09:37:40 +0100 Subject: [PATCH] Fix wrong argument for pooling (forward) --- src/cnn/cnn.c | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/cnn/cnn.c b/src/cnn/cnn.c index ceecbc5..1b46e23 100644 --- a/src/cnn/cnn.c +++ b/src/cnn/cnn.c @@ -121,9 +121,9 @@ void forward_propagation(Network* network) { return; } else { // Pooling sur une matrice if (pooling==1) { - make_average_pooling(input, output, activation, output_depth, output_width); + make_average_pooling(input, output, input_width/output_width, output_depth, output_width); } else if (pooling==2) { - make_max_pooling(input, output, activation, output_depth, output_width); + make_max_pooling(input, output, input_width/output_width, output_depth, output_width); } else { printf("Erreur dans la reconnaissance de la couche de pooling: %d,%d \n", pooling, i); }