augustin64/tipe
1
1
Fork 0
mirror of https://github.com/augustin64/projet-tipe synced 2025-01-23 23:26:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix wrong argument for pooling (forward)

Browse Source
This commit is contained in:
augustin64 2023-03-01 09:37:40 +01:00
parent 0b8ed61a32
commit 95ce123587
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/cnn/cnn.c
View File

@ -121,9 +121,9 @@ void forward_propagation(Network* network) {
return; return;
} else { // Pooling sur une matrice } else { // Pooling sur une matrice
if (pooling==1) { 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) { } 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 { } else {
printf("Erreur dans la reconnaissance de la couche de pooling: %d,%d \n", pooling, i); printf("Erreur dans la reconnaissance de la couche de pooling: %d,%d \n", pooling, i);
} }